[프로그래머스 코딩테스트] lv1. [1차] 비밀지도

문제 설명

비밀지도

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

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

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

 

나의 풀이

def n_bin(num, n):
    answer = []
    while num:
        answer.append(num % 2)
        num = num // 2
    if len(answer) < n:
        answer = answer + [0] * (n - len(answer)) 
    return reversed(answer)

def solution(n, arr1, arr2):
    answer = []
    for a, b in zip(arr1, arr2):
        row = ""
        for x, y in zip(n_bin(a, n), n_bin(b, n)):
            if (x or y) == 1:
                row += "#"
            else:
                row += " "
        answer.append(row)
        
    return answer

나는 n길이의 이진수를 만드는 함수를 직접 작성하여 문제를 해결했다. 이 코드의 문제점은 O(n^2)의 시간이 걸린다는 점이다. 다른 사람들의 풀이를 찾아보면서 간단하면서 직관적인 코드를 발견할 수 있었다. 해당 코드는 다음과 같다.

def solution(n, arr1, arr2):
	answer = []
    for i, j in zip(arr1, arr2):
    	row = str(bin(i|j)[2:])
        row = row.rjust(n, '0')  # n 길이가 될 수 있게 문자열 앞부분에 0을 채워넣어라
        row = row.replace('0', ' ').replace('1', '#')
    	answer.append(row)
    return answer

rjust 함수는 문자열의 길이가 n이 될 수 있게 원하는 character를 넣어줄 수 있게 해주는 함수이다. 이에 대한 자세한 설명은 다음 블로그 글에서 확인해볼 수 있다.

rjust 함수 설명 : https://kkamikoon.tistory.com/136

[Python] 스트링 앞에 0 채우는 방법(zfill, rjust)

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/17681

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