비밀지도

문제

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

 

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 '공백'(' ')' 또는 '벽'('#') 두 종류로 이루어져 있다.

2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 '지도 1'과 '지도 2'라고 하자. 지도 1 또는 지도 2 중 어느 

   하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다. 

3. '지도 1'과 '지도 2'는 각각 정수 배열로 암호화되어 있다.

4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다. 

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

 

제한사항 

· 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]
출력	['#####', '### #', '## ##', ' ####', '#### ']

---

풀이

1. Me

def solution(n, arr1, arr2):
    answer = []
    b_arr1 = []; b_arr2 = []
    
    for i in range(n):
        b_arr1.append([bin(arr1[i])[2:]])
        b_arr2.append([bin(arr2[i])[2:]])    
            
    for i in range(n):
        if len(b_arr1[i]) != n:
            for j in range(n - len(b_arr1[i])):
                b_arr1[i].insert(j, '0')
    
    print(b_arr1)
    # for i in range(n):
    
    return answer

입력받은 정수를 2진수로 변환해야 하는데, 위의 코드와 같이 작성할 경우 2 = 00010(2)와 같이 나오지 않고, 10(2)와 같이 출력되어 길이가 맞지 않는 문제가 발생하였다. 이를 해결하기 위한 코드를 구상하다 실패 .. 

 

2. Others

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

이 코드를 보고 깨달은 것은, 이진수로 전환했을 때 서로의 길이가 다르더라도 비트 연산을 적용하면, 영향받지 않고 정상적인 연산을 수행한다는 것이다. (0b1001, 0b1110)에 or 비트 연산을 적용하면, 0b1111 결과값이 나옴..) 이후 rjust 연산을 적용하여 a12를 오른쪽 정렬하고, a12의 길이가 n보다 작은 경우 맨 왼쪽에 0을 채워준다.. (내가 구현하고 싶었던 부분이 rjust 연산 !!) 이후에는 1을 '#'으로, 0을 ''으로 바꿔주면서 정답을 구했다.

import re

def solution(n, arr1, arr2):
    answer = ["#"] * n
    for i in range(0, n):
        answer[i] = str(bin(arr1[i]|arr2[i]))[2:]
        answer[i] = re.sub('1', '#', '0'*(n-len(answer[i]))+answer[i])
        answer[i] = re.sub('0', ' ', answer[i])
    return answer

re.sub 연산에 대해 좀 더 살펴보고 위 코드를 봐야할 것 같다. 0' * (n-len(answer[i]) + answer[i]) 연산을 통해 구현하고 싶어했던 부분을 구현하였다. (진수 -> 문자열)

 

* re.sub(패턴, 바꿀 문자열, 문자열, 바꿀 횟수)와 같은 형태로 사용하며, 바꿀 횟수를 생략하면 찾은 문자열을 모두 바꿈 

---

자료 정리

1. rjust(), ljust(), zfill()

1) rjust() 연산

-> 오른쪽으로 정렬하도록 도와준다. rjust를 통해 공백의 수, 공백을 메워줄 문자를 넣어준다. 

2) ljust() 연산

-> 왼쪽으로 정렬하도록 도와준다. ljust를 통해 공백의 수, 공백을 메워줄 문자를 넣어준다. 

3) zfill() 연산

-> 정렬은 하지 않고, 0을 왼쪽에 채워준다. 

4) center() 연산 

-> 정의된 문자열을 앞뒤로 특정 문자로 채운뒤, 중앙 정렬 

'77'.rjust(5, '0')   # 00077 
'77777'.rjust(5, '0')   # 77777
'123'.rjust(5, 'a')   # aa123
'123'.rjust(3, 'a')   # 123 

'222'.ljust(3, '0')   # 222
'222'.ljust(10,'a')   # 222aaaaaaa

'2'.zfill(3)   # 002
'50000'.zfill(5)   # 50000
'123'.zfill(5)   # 00123

'I am cori'.center(20, '0')