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

2021년 월 일 Algorithm, 프로그래머스, 비밀지도

2018년 카카오 블라인드 채용 1차 문제

TL;DR

• 비트 연산자를 활용할 수 있는지

문제 분석

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

 

- 1, 3 ~ 4 : 문제의 입력 형태에 대해서 설명하고 있다.

 

이진수 숫자 형태로 암호화된 입력이 주어진다. 실제 지도의 모습은 왼쪽 표와 같다.

 

- 2 : 문제 해결에 중요한 단서를 주고 있다. "지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다." 이 문구는 비트 연산자 OR 연산에 해당하는 내용이다. 이를 활용하면 문제를 보다 쉽게 해결할 수 있다.

입출력 형태

입출력 예시

 

-n : 지도의 크기를 나타낸다. 지도는 문제 조건에 따라 주어진 n 길이의 정사각형 크기이다.

- arr1, arr2 : 암호화된 지도를 나타낸다. 각 원소를 이진수의 형태로 복호화한 후 문제를 해결해야 한다.

- 출력 : 두 지도를 합친, 구해야 하는 답안이다.

풀이

def solution(n, arr1, arr2):
    answer = []
    for a1, a2 in zip(arr1, arr2):
    	code = str(bin(a1 | a2)[2:]) # 두 지도의 각 줄을 복호화한 후 OR 연산을 진행한다.
        if len(code) < n:
        	code = '0' * (n - len(code) + code
        code = code.replace('0', ' ')
        code = code.replace('1', '#')
        answer.append(code)
        
    return answer

 

- OR 연산을 진행하고 나서 슬라이싱을 하는 이유는 그렇지 않을 경우 가장 앞에 이진수임을 나타내는 `0b`가 붙기 때문이다.

- code의 길이와 n의 길이를 비교하는 이유는 bin을 활용하여 이진수로 변환할 경우 딱 해당 자리수에 맞는 형태로 만들기 떄문이다.

- 예 : n = 5일 때 9를 이진수로 변환하면 1001(2)와 같이 된다. 이 경우 앞에 0이 들어가지 않는다. 이를 구현하기 위한 코드이다.

- 문자열 0과 1로 구성되어 있는 것을 문제에서 원하는 출력 형태인 공백와 #으로 바꾸고, 정답 배열에 넣는다.

- 정답을 반환한다.