문제 설명
네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
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을 만족한다.
출력 형식
원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.
입출력 예제
매개 변수 | 값 |
n | 5 |
arr1 | [9, 20, 28, 18, 11] |
arr2 | [30, 1, 21, 17, 28] |
출력 | ["#####", "# # #", "### #", "# ##", "#####"] |
나의 코드
def solution(n, arr1, arr2):
trans_arr1 = []
trans_arr2 = []
answer = []
for x in arr1:
s = format(x, 'b') #2진수로 변환한다.
trans_arr1.append('0'*(n-len(s)) + s) #2진수가 n자리보다 작은 경우 앞에 0으로 채워준다.
for y in arr2:
s = format(y, 'b')
trans_arr2.append('0'*(n-len(s)) + s)
for i in range(n):
num = ''
for j in range(n):
if trans_arr1[i][j] == '0' and trans_arr2[i][j] == '0': #두 arrary가 0이면 공백, 아니면 '#'반환
num += ' '
else: num += '#'
answer.append(num)
return answer
* 10진수에서 2진수, 8진수, 16진수 변환하기
- 각각 bin, oct, hex 함수를 사용하면 된다.
- 그 값만을 추출하고 싶은 경우에는 format 함수를 사용할 수 있다.
다른 풀이
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
- if문을 사용하지 않아도 bin(i|j)로 한번에 계산해줄 수 있다.
- rjust함수를 이용해서 '0'으로 앞 부분을 채워줄 수 있다.
lv.1 끝! 내일부터 젼짱은 lv.2로 레벨업 합니다하ㅏ하하ㅏ
'3. 알고리즘 > 프로그래머스' 카테고리의 다른 글
[python/해시] 25. 전화번호 목록(feat. startswith 함수) (0) | 2021.01.12 |
---|---|
[python] 24. 기능개발 (0) | 2021.01.11 |
[python] 22. 실패율(feat. dictionary 정렬하기) (0) | 2021.01.09 |
[python] 21. 다트 게임(feat. 정규표현식) ★ (0) | 2021.01.08 |
[python] 20. 예산 (0) | 2021.01.08 |