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

비밀지도

문제 설명

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

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

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

제한 조건

• 놀이기구의 이용료 price : 1 ≤ price ≤ 2,500, price는 자연수
• 처음 가지고 있던 금액 money : 1 ≤ money ≤ 1,000,000,000, money는 자연수
• 놀이기구의 이용 횟수 count : 1 ≤ count ≤ 2,500, count는 자연수

풀이

def change(num, n): # 2진수로 변환
    answer = []
    while num:
        answer.append(num % 2)
        num = num // 2
    
    while len(answer) < n: # 자리수 맞추기
        answer.append(0)
    answer.reverse()
    return answer
​
def solution(n, arr1, arr2):
    answer = []
    map1 = [change(a, n) for a in arr1] # 지도 1
    map2 = [change(a, n) for a in arr2] # 지도 2
    
    for m1, m2 in zip(map1, map2):
        row = []
        for i in range(n):
            if m1[i] + m2[i] >= 1: # 둘 중에 하나라도 벽인지 확인
                row.append("#")
            else:
                row.append(" ") # 둘 다 공백인지 확인
        answer.append(''.join(row))
    return answer

다른 사람 풀이

각 기능을 직접 구현하지 않고 빌드인 함수를 쓰는 방법이 있었다.. !

bin 함수를 사용하면 integer를 binary로 바꿀 수 있다. 심지어 비트 연산을 해서 넣어도 된다.

>>> bin(10)
'0b1010'
>>> bin(5|2)
'0b111'

rjust, ljust 오른쪽, 왼쪽으로 정렬해 공백의 수, 공백을 메워줄 문자를 넣어준다.

val  = "77".rjust(5, "0")
>>> 00077

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