반응형
문제
네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
- 지도는 한 변의 길이가
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
≦ 2^n – 1을 만족한다.
출력형식
원래의 비밀지도를 해독하여 "#"
, 공백
으로 구성된 문자열 배열로 출력하라.
입출력 예제
풀이
def solution(n, arr1, arr2):
answer = []
arr1 = (list(map(lambda x:format(x,'b'),arr1)))
arr2 = (list(map(lambda x:format(x,'b'),arr2)))
row=''
for i in range(n):
arr = str(int(arr1[i])+int(arr2[i])).zfill(n)
for j in arr:
if j == '0':
row+=' '
else:
row+='#'
answer.append(row)
row=''
return answer
배열 arr1과 배열 arr2의 값을 2진수로 바꿔서 저장한다.
배열 arr1과 배열 arr2의 값들을 더해주고 arr에 저장한다. arr의 길이가 n보다 작으면 앞자리를 0으로 채워준다.
arr 에서 각 자리의 숫자가 0이면 ' ', 1이면 '#'으로 바꿔서 리스트 answer에 저장한다.
위의 과정을 n번 반복하고 리스트 answer을 리턴한다.
format()
- Python 내장함수인 format()은 입력받은 숫자를 접두어를 제외하고 다른진수의 문자열로 바꿀 수 있다.
>>> format(42, 'b') '101010' >>> format(42, 'o') '52' >>> format(42, 'x') '2a' >>> format(42, 'X') '2A' >>> format(42, 'd') '42'
- 두번째 인자 앞에
#
를 붙이면 접두어를 포함시키는 것도 가능하다. >>> format(42, '#b') '0b101010' >>> format(42, '#o') '0o52' >>> format(42, '#x') '0x2a' >>> format(42, '#X') '0X2A'
zfill()
- zfill() 메서드를 사용하면 문자열이 특정 길이가 되도록 문자열 앞에 0을 채울 수 있다.
참조
반응형