문제 설명
네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
- 지도는 한 변의 길이가 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을 만족한다.
출력 형식
원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.
문제 풀이
보물지도가 2진수로 구현되어 있지만 이진수의 합을 구하고 변환하는 문제는 아니다.
벽과 벽이 없는 곳을 1과 0으로 표현하기 위해서 라는 이유로 2진수를 이용했다고 보는 편이 맞다.
문제에서 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이라고 했으므로 그 점만 유의하여 코드를 구현하면 된다.
풀이 과정
arr1과 arr2가 2진수가 아닌 10진수로 표현되었기 때문에 먼저 이를 2진수로 변환해준다.
arr1_bi = []
arr2_bi = []
for i in range(n):
arr1_bi.append(bin(arr1[i])[2:])
arr2_bi.append(bin(arr2[i])[2:])내장함수인 bin 함수를 사용하였다. bin 함수는 10진수를 2진수로 변환해주는 함수이다.
사용 예를 간단히 정리해본다면
a = 13
b = 23
print(bin(a), bin(b))
>>> 0b1101 0b10111주어진 매개변수를 2진수로 변환하면서 앞에 binary를 의미하는 '0b'가 붙는다.
따라서 이진수만 뽑아내기 위해서는 앞의 '0b'를 제외하기 위해 인덱싱하여 사용했다.
이 뒤에서 조금 고민이 생겼는데, 보물지도 형태라면 00010과 같이 앞자리에도 0이 포함되도록 데이터를 조작해야 했다.
처음 문제를 풀 당시에 rjust나 zfill 함수를 생각하지 못했기 때문에 조금 무식한 방법으로 접근했다.
for j in range(n):
for k in range(1, n):
if len(arr1_bin[j]) == k:
arr1_bin[j] = ('0' * (n-k)) + arr1_bin[j]
for q in range(1, n):
if len(arr2_bin[j]) == q:
arr2_bin[j] = ('0' * (n-q)) + arr2_bin[j]지도의 변의 길이에 해당하는 값이 주어졌으니
arr1_bin과 arr2_bin에 들어있는 값 중에 변의 길이만큼 충족되지 않는 이진수의 앞에 str 형태의 0을 추가하는 코드를 구현했다.
여기까지의 과정을 통해 arr1과 arr2의 배열을 이진수로 변환하는 것까지 완료했다.
변환된 배열을 바탕으로 최종 보물지도의 형태를 배열로 나타내기 위한 코드를 구현했다.
for idx in range(n):
tmp = ''
for p in range(n):
if arr1_bin[idx][p] == '1' or arr2_bin[idx][p] == '1':
tmp += '#'
elif arr1_bin[idx][p] == '0' and arr2_bin[idx][p] == '0':
tmp += ' '
answer.append(tmp)주의해야 할 점이 있다면 모든 이진수가 str형태로 되어있다는 점이다.
따라서 arr1_bin과 arr2_bin 둘 중 하나에 '1'이 포함된다면 '#'을, 그렇지 않으면 공백으로 처리하여 보물지도 배열을 완성하였다.
최종 코드는 다음과 같다.
def solution(n, arr1, arr2):
answer = []
arr1_bin = []
arr2_bin = []
for i in range(n):
arr1_bin.append(bin(arr1[i])[2:])
arr2_bin.append(bin(arr2[i])[2:])
arr1_bin[i] = ('0' * (n-len(arr1_bin[i]))) + arr1_bin[i]
arr2_bin[i] = ('0' * (n-len(arr2_bin[i]))) + arr2_bin[i]
tmp = ''
for p in range(n):
if arr1_bin[i][p] == '1' or arr2_bin[i][p] == '1':
tmp += '#'
elif arr1_bin[i][p] == '0' and arr2_bin[i][p] == '0':
tmp += ' '
answer.append(tmp)
return answer실행 결과
채점 결과
