Lv.1 [1차] 비밀지도

문제

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

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

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

예제

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

1 ≦ n ≦ 16
arr1, arr2는 길이 n인 정수 배열로 주어진다.
정수 배열의 각 원소 x를 이진수로 변환했을 때의 길이는 n 이하이다. 즉, 0 ≦ x ≦ 2n - 1을 만족한다.
출력 형식
원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.

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

나의 풀이

format(x, 'b')를 사용하여 arr1과 arr2 리스트의 정수를 2진수 문자열로 변경한 후 zfill(n)으로 문자열의 길이를 n만큼으로 변경한다. 그렇게 이진수의 값으로 만들어진 new_a1, new_a2의 인덱스를 사용하여 같은 위치의 두 값 중 하나라도 '1'이면 '#'을, 두 값이 모두 '0'이면 공백을 더해 새로운 문자열을 만든다. 그렇게 만든 문자열 한 줄line을 answer 리스트에 append하여 정답을 도출했다.

def solution(n, arr1, arr2):
    new_a1 = [format(arr1[i],'b').zfill(n) for i in range(n)]
    new_a2 = [format(arr2[i],'b').zfill(n) for i in range(n)]
    answer = []
    for i in range(n):
        line = ''
        for j in range(n):
            if new_a1[i][j] == '1' or new_a2[i][j] == '1':
                line += '#'
            elif new_a1[i][j] == '0' and new_a2[i][j] == '0':
                line += ' '
        answer.append(line)
        
    return answer

다른 풀이

새로운 개념으로 좀 더 간단하게 구성된 코드가 있어서 참고하려고 한다.
여기서는 비트 OR 연산자를 활용하여 리스트를 2진법으로 변환하고 rjust(n,'0')을 사용해서 자릿수를 n의 길이로 맞췄다. 비트연산자 사용 시 [2:]를 해준 이유는 2진법으로 숫자를 변환하면 '0b11011' 같은 형태로 반환되기 때문에 앞에 0b 부분은 자르고 rjust를 사용하여 빈자리를 0으로 채워주는 것이다. 이 후에는 '1'을 '#'으로, '0'을 공백으로 replace하고 새 리스트에 append하여 답을 도출했다.

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

비트 OR 연산자를 떠올릴 수 있었다면 아래와 풀었을 것 같다.

def solution(n, arr1, arr2):
	answer = []
    for i in range(n):
    	line = format(arr1[i] | arr2[i], 'b').zfill(n)
        line = line.replace('1', '#').replace('0', ' ')
    	answer.append(line)
    return answer

끄적끄적

format()

format()은 숫자 또는 값을 원하는 형식의 문자열로 변환해주는 함수이다. 특히 진수로 변환할 때 많이 사용된다.

# 기본형
format(value, format_spec)

# 주요 포맷 코드
# 2진수
format(10, 'b')

# 8진수
format(10, 'o')

# 16진수 (소문자)
format(255, 'x')

# 16진수 (대문자)
format(10, 'X')

zfill()

zfill()은 문자열의 길이를 지정한 크기만큼 맞추고, 부족한 부분을 0으로 채워주는 문자열 메서드이다.

str.zfill(width)

비트 OR 연산

bin(i|j)[2:]에서 |는 비트 OR 연산자이다. 비트 연산자 동작은 아래와 같다.

9 (1001)
14 (1110)
9 | 14 -> 1111 (2진수)

rjust()

rjust()는 문자열 정렬 함수이다. 문자를 오른쪽으로 정렬한다.
문장열의 길이가 width보다 짧을 경우, 왼쪽에 fillchar를 채워서 문자열의 길이를 맞춘다.

# 기본형
str.rjust(width, fillchar=' ')