[파이썬] 카카오 블라인드 공채 1번 문제

친구따라서 풀어본 카카오 알고리즘 문제였다. 7문제가 출제되었고 부분 점수는 없었다. 나같은 경우에는 1번 문제, 5번 문제만 손댈 수 있었는데 1번 문제는 완벽하게 맞았다. 겨우 한 문제 맞았지만, 이정도라면 충분히 만족(?)스럽다. 꾸준하게 나아가자.

문제

def solution(n, arr1, arr2):
  answer = []
  return answer

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

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을 만족한다.

출력 형식

원래의 비밀지도를 해독하여 '#', 공백으로 구성된 문자열 배열로 출력하라.

입출력 예제

나의 풀이

착안점

테스트 코드

print(solution(n=6, arr1=[46, 33, 33 ,22, 31, 0], arr2=[27 ,56, 19, 14, 14, 0]))

1. 십진수를 이진수로 변환한 뒤, or 연산이 필요했다.

   • 파이썬에는 bin 함수가 있다.
   • bin 함수내에서 or 연산과 and 연산이 가능했다.
   • or은 |로, and는 &로 가능하다.
   • bin(i | j) 로 가능하다

2. 동일한 길이로 주어지는 arr1과 arr2를 이차원 배열로 만들고자 했다.

   • matrix라는 변수로 빈 리스트를 만들어주었다.
   • extend 함수를 써서 리스트 자체가 들어가도록 했다.
     • 즉, [[9, 20, 28, 18, 11], [30, 1, 21, 17, 28]] 이런 형식으로 들어온다.

3. 이렇게 하면 zip(*) 를 사용해서 같은 위치에 있는 인덱스를 활용할 수 있다.

   • 아래 출력결과를 이진수연산을 하면 된다.
     출력결과

   46 27
   33 56
   33 19
   22 14
   31 14
   0   0
   

첫번째 시도

def solution(n, arr1, arr2):
    answer = []
    matrix = []
    matrix.extend((arr1, arr2))
    result = []
    for i, j in zip(*matrix):
        str_bin = bin(i | j)[2:]
        if len(str_bin) != n: # 길이가 n이 아니면 0을 채워줘야 한다.
            while len(res2) != n:
                res2 += str_bin
                res2.insert(0, "0")
            result.append("".join(res2))
        else:
            result.append(bin(i | j)[2:])

    for array in result:
        res = ""
        for j in array:
            res += "#" if j == "1" else " "
        answer.append(res)
    return answer

이 코드는 시간복잡도 O(n^2)이였기 때문에 시간초과 걸리는 테스트 코드가 두 개 있었다. 어떻게든 while문을 없애야 했다.

두번째 시도

서브 함수를 만들어도 그 안에는 while문이 존재했기 때문에 이 방법은 소용이 없었다.
나는 구글에 도움을 요청하기로 했다. python str fill with 0라고 검색을 했고,
무릎을 탁 치는 검색결과를 얻을 수 있었다.

이 스택오버플로 글을 보자마자 왠지 될 것만 같았다.
zfill(width) 에 있는 width값만큼 0을 채워주는 함수다.

print("5".zfill(5))
>>> 00005

def solution(n, arr1, arr2):
    answer = []
    matrix = []
    matrix.extend((arr1, arr2))
    result = []
    for i, j in zip(*matrix):
        str_bin = bin(i | j)[2:] # bin(i) 출력결과는 0b001100 이런 형식으로
                                 # 나오기 때문에 2번째부터 출력을 유도했다.
        if len(str_bin) != n: # 길이가 n이 아니면 0을 채워줘야 한다.
            result.append(str_bin.zfill(n)) # string.zfill(width)

        else:
            result.append(bin(i | j)[2:])

    for array in result:
        res = ""
        for j in array:
            res += "#" if j == "1" else " "
        answer.append(res)
    return answer

테스트 코드

print(solution(n=5, arr1=[9,20,28,18,11], arr2=[30,1,21,17,28]))
print(solution(n=6, arr1=[46, 33, 33 ,22, 31, 0], arr2=[27 ,56, 19, 14, 14, 0]))
print(solution(n=7, arr1=[46, 33, 33 ,22, 31, 50, 31], arr2=[27,56, 19, 14, 14, 10, 31]))

새로 배운 메서드

zfill(width) : 글자 맨 앞에 width 길이만큼 0을 채워준다.

"5".zfill(5)
>>> 00005

bin()함수 안에서의 연산

print(bin(5 & 8)) # AND 연산 (모두 True(1)일때만 True(1)이 되는 연산)
>>> 0b0

print(bin(5 | 8)) # OR 연산 (모두 False(0)일떄만 False(0)이 되는 연산)
>>> 0b1101