FrontEnd
Javascript
Diary
ML
CS
Django
Algorithm
AWS
Co-Work
HTML
CSS
Python
React
ReactNative

#57 알고리즘 연습 - [1차] 비밀지도 Python

문제 설명

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

지도는 한 변의 길이가 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]
출력 [”######”, ”### #”, ”## ##”, ” #### ”, ” #####”, ”### # ”]

문제풀이 일단 주어진 10진수 수를 2진수로 바꾸어줘야한다. 그런데 왠지 2진수로 변환하는 과정은 module을 불러와야 할 것 같았기에 하드코딩을 해주었다. 그리고 list안에 들어간 이진수의 1과 0이 각각 벽과 길을 의미했기때문에 이진수의 각 자리를 탐색하기위해 문자열로 이진수를 만들어 주는게 keypoint이다.

내 풀이 🏆

def convert_2(n,arr):
    arr_conv_2 = []
    for i in arr:
        tmp = ""
        for ex in range(n-1,-1,-1):
            if i//(2**ex) == 1:
                tmp += "1"
            else:
                tmp += "0"
            i = i%(2**ex)

        arr_conv_2.append(tmp)
    return arr_conv_2

def solution(n, arr1, arr2):
    answer = []
    arr1_bin = convert_2(n,arr1)
    arr2_bin = convert_2(n,arr2)
    for i in range(0,n):
        tmp_row = ""
        for j in range(0,n):
            if int(arr1_bin[i][j]) or int(arr2_bin[i][j]):
                tmp_row += "#"
            else:
                tmp_row += " "
        answer.append(tmp_row)

    return answer
  • 나름 각 자리를 int로 바꾸어 or연산자를 잘 이용해 if를 단순화 시켰다고 생각했는데 일단 이진법을 바꿔주는 내장함수를 사용 안했다는 것과 for문을 2번 사용했다는게 조금 단점이다.

다른 풀이 🏆

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
  • zip() 내장함수를 이용해서 [1,2,3], [4,5,6]이 있을때 [(1,4),(2,5),(3,6)]과 같은 방법으로 묶은다음 i와 j를 이용했다.
  • 그런데 이때 각 숫자들에 대해 일단 i | j를 이용해서 비트연산자를 적용해줬는데 이렇게 해줄시 i 와 j에 대해서 각각 2진수로 변환하게 되고 2진수의 각 자리에 대해서 or 비트 연산을 수행해주게 된다. 그리고 그 값에 대해 다시 10진수로 표현해주는데 이를 bin이라는 내장함수를 이용해서 0b가 앞에 붙은 2진수로 변환시켜준다. 앞에 0b가 붙어 있기에 index2부터 끊어서 string으로 돌려 주었다.
  • 무엇보다도 a12는 i와 j의 비트연산을 거친 뒤에 반환되는 값이므로 주어진 n자리만큼의 값이 아닐 수있다. 따라서 rjust라는 함수를 이용해 앞에 0을 자릿수만큼 채워주는데 rjust(목표 길이, [채우고자 하는 문자]) 의 방법으로 사용할 수 있다.

python의 내장함수중 bin의 존재와 rjust, 비트연산자에 대해서 배울수 있었던 좋은 풀이였다. 어떻게 보면 or연산자와 이진법 변환 함수를 이용하여 나는 위의 코드를 하드코딩했었는지도 모르겠다😥