[프로그래머스/PYTHON] [1차] 비밀지도

문제 링크

코딩테스트 연습 - [1차] 비밀지도

---

● 문제 설명

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

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.
2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.

 

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

---

● 문제 해결 방향

두 배열의 각 숫자들을 한 짝씩 2진수로 바꾸어 문자열로 전환해주고,

두 문자를 비교하여 각 자리에 적절한 문자를 넣고자 했다.

---

● 문제 해결 코드

 

처음 짠 코드는 다음과 같다.

def solution(n, arr1, arr2):
    answer = []
    for i in range(n):
        word = ''
        
        n1 = str(bin(arr1[i]))[2:] 
        n1 = n1 if arr1[i] > 2**(n-1) else '0'*(n - len(n1)) + n1
        n2 = str(bin(arr2[i]))[2:]
        n2 = n2 if arr2[i] > 2**(n-1) else '0'*(n - len(n2)) + n2
        
        for pair in zip(n1, n2):
            if '1' in pair:
                word += '#'
            else :
                word += ' '
        answer.append(word)
    return answer

우선 각 배열의 숫자를 bin() 함수를 통해 2진수로 바꾸어주었다.

그리고 글자를 한 글자씩 비교하기에는 문자열이 더 적절하다고 판단해 문자열로 바꾸어주었다.

 

숫자에 따라 자릿수가 바뀌기 때문에 n과 2진수의 길이를 통해 전체 2진수의 길이를 통일시켜 주었다.

 

zip을 통해 각 자리마다 1이 포함되어있으면 '#'을, 포함되어있지 않으면 ' '을 넣어

answer에 append 해주었다.

 

문자열을 조작하는 부분에서 코드가 길어지고 복잡해진 것 같아 마음에 걸리는 부분이 있었는데

이 부분에 대해서 다른 분들이 제시하신 풀이를 보고 바로 해소되었다.

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

내가 찝찝하게 생각했던 부분 두군데를 정확히 내장 함수만으로 구현한 코드이다.

먼저,

나는 각자리에 or 연산을 for문과 zip함수를 통해서 실행했는데 

여기서는 str( bin ( i | j ) )으로 각 자리에 대해서 or 연산을 바로 실행해버렸다.

그리고 n 과 2진 문자열의 길이를 비교해 전체 길이를 통일시켜준 부분을 rjust(zfill)을 통해 해결했다.

---

● 알게 된 것 / 아쉬운 점

level 1단계 문제는 항상 최소한으로 코드를 작성해서 해결하려고 노력하고 있다.

그걸 위해서 많은 내장함수를 알고 있는 것이 필요하다고 생각한다.

 

오늘은 유용한 연산방식과 zfill, rjust라는 유용한 내장 함수에 대해서 알게 되었다.