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

• [1차] 비밀지도

문제 설명

비밀지도

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

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

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

내 코드

def solution(n, arr1, arr2):
    answer = []

    arr1_2d = transform(n, arr1)
    arr2_2d = transform(n, arr2)
    
    for i in range(n):
        arra = arr1_2d[i]
        arrb = arr2_2d[i]
        temp = arra
        for j in range(n):
            if arra[j] == "#" or arrb[j] == "#":
                temp = temp[:j] + "#" + temp[j+1:]
        answer.append(temp)
        
    return answer

def transform(n, arr):
    result = []
    for i in arr:
        result.append(format(i, 'b'))
    for i in range(n):
        if len(result[i]) != n:
            result[i] = '{0:0>{1}}'.format(result[i], n)
    for i in range(n):
        result[i] = str(result[i]).replace('1', '#')
        result[i] = str(result[i]).replace('0', ' ')
            
    return result

 

다른 사람 코드

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

• arr1과 arr2를 or연산을 이용하여 2진수로 바꾼다. bin()는 앞에 0b가 붙어서 [2:]로 없애준다.
• 문자열을 오른쪽으로 붙이고 rjust()로 크기 n에 맞게 0으로 채워준다.
• 그리고 1은 #으로, 0은 공백으로 바꾸어준다...

rjust와 or연산, zip을 기억해두어야 겠다