[프로그래머스] 비밀 지도 python

문제

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

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

풀이

1. 십진수 -> 이진수 변환하는 방법을 알아야 합니다.

def solution(n, arr1, arr2):
    answer = []
    first_map = [[] for _ in range(n)]
    second_map = [[] for _ in range(n)]
    
    for i, num in enumerate(arr1):
        while num > 0:
            first_map[i].append(num%2)
            num = num // 2
        first_map[i].reverse()
        
        if len(first_map[i]) != n:
            tmp = n-len(first_map[i])
            for _ in range(tmp):
                first_map[i].insert(0, 0)
    for i, num in enumerate(arr2):
        while num > 0:
            second_map[i].append(num%2)
            num = num // 2
        second_map[i].reverse()

        if len(second_map[i]) != n:
            tmp = n-len(second_map[i])
            for _ in range(tmp):
                second_map[i].insert(0, 0)

    #print(first_map)
    #print(second_map)

    for i in range(n):
        res = ""
        for j in range(n):
            if first_map[i][j] == 1 or second_map[i][j] == 1:
                res += "#"
            elif first_map[i][j] == 0 and second_map[i][j] == 0:
                res += " "
        answer.append(res)
    return answer

 

코딩테스트 연습 - [1차] 비밀지도