[Algorithm] 프로그래머스 - [1차] 비밀지도

시작하며...

모든 알고리즘 문제 풀이는 제가 직접 짜서 정답을 맞춘 결과만을 공유합니다.

마지막 'More Solution'은 다른 정답자들 풀이 중 생각지 못했던 부분들이나 좋게 느껴진 풀이법 몇개를 가져와서 공유하였습니다.

 

[Level 1] [1차] 비밀지도

프로그래머스

 

문제 설명

비밀지도

네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.
2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.

네오가 프로도의 비상금을 손에 넣을 수 있도록, 비밀지도의 암호를 해독하는 작업을 도와줄 프로그램을 작성하라.

입력 형식

입력으로 지도의 한 변 크기 n 과 2개의 정수 배열 arr1, arr2가 들어온다.

• 1 ≦ n ≦ 16
• arr1, arr2는 길이 n인 정수 배열로 주어진다.
• 정수 배열의 각 원소 x를 이진수로 변환했을 때의 길이는 n 이하이다. 즉, 0 ≦ x ≦ 2n - 1을 만족한다.

출력 형식

원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.

입출력 예

 

My Solution

def solution(n, arr1, arr2):
    answer = []
    bin_arr1 = [x[2:].zfill(n) for x in map(bin, arr1)]
    bin_arr2 = [x[2:].zfill(n) for x in map(bin, arr2)]
    
    for bin1, bin2 in zip(bin_arr1, bin_arr2):
        tmp = ''
        for i in range(n):
            if bin1[i] == '1' or bin2[i] == '1':
                tmp += '#'
            else:
                tmp += ' '
        answer.append(tmp)
    return answer

 문제 풀이 설명!

• 문제에서 주어진 각각의 배열(arr1, arr2)을 map을 활용하여 2진수로 변환 : bin()
• bin() 메소드를 이용하여 2진수로 변환하면 0x____ 형태로 바뀌기 때문에 순수한 2진수 숫자를 얻기위해 문자열 인덱싱 활용 - x[2:]
• 변환된 2진수가 숫자에 따라 길이가 6개가 되지 않기때문에, 내장함수를 이용하여 길이 n만큼으로 나머지는 0을 채워주기 - zfill()
• 2진수 변환된 배열(bin_arr1, bin_arr2)의 각각의 요소들을 비교 - 이중 for문
• 두 배열의 i 번째 인덱스 숫자 중에 1이 하나라도 있으면( 'or' ) 문자열 변수 tmp에  '#' 을 추가 - bin1[i] == '1' or bin2[i] == '1'
• 그렇지 않으면 공백 추가
• 한줄이 다 변환되면 tmp 변수를 answer에 추가
• answer 리턴!

 

More Solution

def solution(n, arr1, arr2):
    answer = []
    for i,j in zip(arr1,arr2):
        a12 = str(bin(i|j)[2:])
        a12=a12.rjust(n,'0')
        a12=a12.replace('1','#')
        a12=a12.replace('0',' ')
        answer.append(a12)
    return answer

다른 풀이법에서는 나와는 다른 방법을 몇가지 사용하여서 가져와보았다.

우선 비트 연산자를 활용해서 각 요소에 벽(1)이 있는지를 판단하고 그 값을 bin() 메소드로 이진수 변환

2진변환 후 빈 자리를 채우는 방법으로는 zfill 대신 rjust 메소드를 사용

합쳐진 2진수 배열을 replace를 통해 1은 #으로, 0은 공백으로 변환