[Programmers] [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≦ 2^n – 1을 만족한다.
출력 형식
원래의 비밀지도를 해독하여 "#", 공백으로 구성된 문자열 배열로 출력하라.
입출력 예제
| 매개변수 | 값 |
|---|---|
| n | 5 |
| arr1 | [9, 20, 28, 18, 11] |
| arr2 | [30, 1, 21, 17, 28] |
| 출력 | ["#####","# # #", "### #", "# ##", "#####"] |
| 매개변수 | 값 |
|---|---|
| n | 6 |
| arr1 | [46, 33, 33 ,22, 31, 50] |
| arr2 | [27 ,56, 19, 14, 14, 10] |
| 출력 | ["######", "### #", "## ##", " #### ", " #####", "### # "] |
풀이
두 비밀지도를 겹쳤을 때 한 지도에서라도 벽이 있을 경우 그 구역은 벽으로 표시해야 한다. 즉, or 연산을 써야 한다는 것을 빠르게 캐치했다.
처음 제출 코드
def solution(n, arr1, arr2): result = [] for i,j in zip(arr1,arr2): target = str(bin(i|j))[2:].zfill(n) temp = '' for t in target: temp = temp + '#' if t=='1' else temp+' ' result.append(temp) return result | cs |
통과는 했지만 지저분한게 느껴졌고 2중 for문이 있는 것이 거슬렸다. 다른 사람의 풀이에서 replace 메서드를 사용하면 for문을 사용하지 않아도 된다는 것을 알아내어 바로 수정했다.
최종 코드
def solution(n, arr1, arr2): result = [] for i,j in zip(arr1,arr2): result.append( str(bin(i|j))[2:].zfill(n).replace('1','#').replace('0',' ')) return result | cs |
줄여서 간편하게 쓸 수 있는 부분을 한 줄로 합쳤다.
list comprehension을 사용해 한줄 코딩도 해봤다.
def solution(n, arr1, arr2): return [str(bin(i|j))[2:].zfill(n).replace('1','#').replace('0',' ') for i,j in zip(arr1,arr2)] | cs |
No comments: