코딩테스트 연습 > 카카오 2018 블라인드 1차
비트 연산을 할 수 있는지 평가하는 문제.
두 개의 리스트를 2진수로 변환하여 OR 연산을 하고, 각 원소를 특정 문자로 변환하는 문제였다.
def solution(n, arr1, arr2):
answer = []
for n1, n2 in list(zip(arr1,arr2)):
bin_str = bin(n1 | n2)[2:].zfill(n)
bin_str = bin_str.replace('1','#')
bin_str = bin_str.replace('0',' ')
answer.append(bin_str)
return answer
arr1, arr2 배열을 2진수로 이루어진 배열로 만들어두고, 2진수의 각 자리수마다 OR 연산을 해서 n^2번 계산했던 풀이를 zip 함수와 bin 함수를 이용하여 n번의 계산으로 개선했다. 각 원소의 개수가 같고 해당 원소들을 이용해 반복문을 만드는게 필요하다면 zip 함수를 이용하는게 편리한 것 같다. 그리고 비트 연산을 이용하면 OR 연산을 반복할 필요가 없다는 것을 배웠다.
첫 구현에서는 2진수의 각 자리수별로 OR 연산을 하도록 구현했는데 비효율적이라는 생각이 들었다. 파이썬에서는 '|' 연산자로 쉽게 비트 연산을 할 수 있었다. 그리고 bin이라는 함수를 이용하면 10진수를 2진수로 바꾸어 바로 비트 연산을 할 수 있기 때문에 편리했다.
주의해야 할 점은 bin 함수는 앞에 무조건 0b라는 문자가 붙어서 슬라이싱을 해주어야 하고, 맨 앞자리가 0으로 시작하는 경우가 있기 때문에 자리수를 맞춰서 맨 앞에 0을 채워주어야 한다.
네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.
전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
"지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.
매개변수 | 값 |
---|---|
n | 5 |
arr1 | [9, 20, 28, 18, 11] |
arr2 | [30, 1, 21, 17, 28] |
출력 | ["#####","# # #", "### #", "# ##", "#####"] |