[프로그래머스] : [1차] 비밀지도

문제

https://school.programmers.co.kr/learn/courses/30/lessons/17681

프로그래머스

 

문제 설명

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

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

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

 

입력 형식

입력으로 지도의 한 변 크기 n 과 2개의 정수 배열 arr1, arr2가 들어온다.

• 1 ≦ n ≦ 16
• arr1, arr2는 길이 n인 정수 배열로 주어진다.
• 정수 배열의 각 원소 x를 이진수로 변환했을 때의 길이는 n 이하이다. 즉, 0 ≦ x ≦ 2n - 1을 만족한다.

 

출력 형식

원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.

 

입출력 예제

 

풀이

이 문제는 십진수를 이진수로 변환해서, 비트 연산자를 사용해 풀면 된다.

1. 각 배열의 숫자를 이진수로 변환하고

2. 변환한 이진수끼리 비트 연산자 | 를 사용해 둘 다 0인 경우만 0이 되도록 만든다.

이 때 | 연산자는 (OR) 연산자로, '1100' | '1010' => '1110' 으로 계산된다.
같이 알면 좋은 & 연산자는 (AND) 연산자로, '1100' & '1010' => 1000' 으로 계산된다.
즉, |는 둘 중 하나만 1이더라도 1로 계산되고,
&는 둘 다 1이어야만 1로 계산된다.

3. 비트 연산자로 변환된 수가 1->'#', 0-> ' ' 으로 변환해준다.

 

 

정답 코드

위 순서대로 구현하면 다음과 같이 구현할 수 있다.

# 최종 결과를 만드는 메인 함수
def solution(n, arr1, arr2):
    # arr1과 arr2를 비트 OR 연산 후 이진 문자열 리스트로 변환
    result = binary_num(arr1, arr2, n)
    answer = []

    # 각 이진 문자열을 '#' 또는 ' '로 변환
    for i in range(len(result)):
        ans = ''
        for j in range(len(result[i])):
            if result[i][j] == '1':  # '1'이면 벽이므로 '#'
                ans += "#"
            else:                   # '0'이면 공백
                ans += " "
        answer.append(ans)  # 완성된 문자열을 리스트에 추가
    return answer

# 두 리스트의 각 요소를 비트 OR 연산한 후, n자리 이진 문자열로 변환하는 함수
def binary_num(arr1, arr2, n):
    result = []
    for a, b in zip(arr1, arr2):  # arr1과 arr2에서 같은 인덱스끼리 짝지어 반복
        binary = bin(a | b)[2:].zfill(n)  # 비트 OR 후, 앞의 '0b' 제거하고 n자릿수로 채움
        result.append(binary)  # 변환된 이진 문자열을 결과 리스트에 추가
    return result

 

이 때 새롭게 알게된 함수는 .zfill()이다. 이 함수는 n자리에서 빈 곳을 채워주는 함수이다.

bin 함수를 잘 사용하는 것과 .zfill()과 같이 자릿수를 고려하는 것이 중요한 핵심이었던 것 같다.