알고리즘 저장소
Programmers Level 1. 2018 KAKAO BLIND RECRUITMENT 비밀지도 (Python)
Diger
2022. 7. 24. 03:06
728x90
문제 설명
비밀지도
네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
지도는 한 변의 길이가 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을 만족한다.
출력 형식
원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.
입출력 예제
매개변수 | 값 |
---|---|
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] |
출력 | ["######", "### #", "## ##", " #### ", " #####", "### # "] |
풀이
import re
def solution(n, arr1, arr2):
or_operation = list()
for i in range(n):
or_operation.append(arr1[i] | arr2[i])
for i in range(n):
or_operation[i] = bin(or_operation[i])[2:n+2]
if len(or_operation[i]) < n:
or_operation[i] = ('0' * (n - len(or_operation[i]))) + or_operation[i]
for i in range(n):
or_operation[i] = re.sub('1', "#", or_operation[i])
or_operation[i] = re.sub('0', " ", or_operation[i])
return or_operation
변수
or_operation : 문제의 요구사항을 알게되면, OR 비트 연산을 해야하는 것을 알게 될 것이다.
그래서 이 OR 연산을 해준 값들을 담을 리스트를 선언했다.
For-Loop-1
첫번째 반복문에서는, 입력받은 배열의 원소를 하나씩 OR 연산을 해주어서, 앞서 선언한 리스트에 담는다.
For-Loop-2
두번째 반복문에서는, 첫번재 반복문에서 얻은 결과를 2진수 값으로 변환해주는 역할을 한다.
이때, 파이썬에서 bin() 을 통해 2진수로 변환을하면, "0b" 라는 추가 수식어가 붙기 때문에 이를 제거하여 리스트에 새로 값을 할당한다.
또한, 계산된 2진수의 자리수가 n보다 작을때는 자리수를 채워주기 위해 맨 좌측에 '0' 으로 자릿수를 채워 주도록 한다.
For-Loop-3
마지막으로 or_operation 에 들어있는 값들을, '1'은 '#' 으로, '0',은 ' ' 으로 변환하여
요구하는 출력형태로 변환한다.
Return
728x90