[프로그래머스] 1차 비밀지도 파이썬

프로그래머스 - 1차 비밀지도

문제

• 한 변의 길이가 공백(" ") 또는 벽("#") 두 종류로 이루어져있는 정사각형 배열이 주어진다.
• 두 지도가 주어져서 두 지도를 합쳤을 때, 공백인 부분은 여전히 공백이나 하나의 지도라도 벽으로 표시된 부분이 있다면 이 부분은 합쳤을 때 벽이 된다.
• 암호화된 배열을 이진수로 바꿨을 때, 벽은 1, 공백은 0에 해당한다.
• 문제에서 주어진 지도를 해석하라.

코드

def convert_binary(n, arr):
    return [bin(num)[2:].zfill(n) for num in arr]


def solution(n, arr1, arr2):
    arr1 = convert_binary(n, arr1)
    arr2 = convert_binary(n, arr2)

    array = [str(int(arr1[i]) + int(arr2[i])).zfill(n) for i in range(len(arr1))]

    result = []
    for x in array:
        tmp = ""
        for k in x:
            if k != "0":
                tmp += "#"
            else:
                tmp += " "

        result.append("".join(tmp))

    return result

설명

• 주어진 두개의 배열을 각각 이진수로 변환 된 문자열로 변경한다. n의 길이에 맞춰서 앞에 0을 채워넣어야하는데, 이때 내장 함수인 zfill을 사용하면 쉽게 문자열의 앞에 0을 채워넣을 수 있다.
• 이진수로 변환된 두 문자열을 더한다. 이때 int로 캐스팅한 후 합연산을 수행하지 않으면 0100111110과 같이 옆으로 붙어버리니 int로 캐스팅해준 후 연산된 값을 다시 str으로 캐스팅해주자. 두 이진수의 값이 각각 01001과 11110의 경우 합을 했을 때, 12111과 같이 2가 생기는데 괜찮다. 어차피 0은 공백이고 두 지도를 합했을 때, 0 + 0 = 0이기 때문에 0이 아니면 벽으로 생각하면 되기 때문이다.
• 연산의 결과값을 반복문을 수행하며 0일 경우 #을, 0이 아닌 경우 " "을 넣은 임시 문자열 tmp를 생성해준다.
• tmp를 join해주어 하나의 값으로 만들어주면 된다.