[프로그래머스-Lv1] [1차] 비밀지도 / Python

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

import sys

input = sys.stdin.readline
n = int(input())
arr1 = list(map(int, input().split()))
arr2 = list(map(int, input().split()))


def solution(n, arr1, arr2):
    answer = []
    for i in range(len(arr1)):
        temp1 = bin(arr1[i] | arr2[i])[2:]
        temp2 = "0" * (n - len(temp1)) + temp1
        temp2 = temp2.replace("0", " ")
        temp2 = temp2.replace("1", "#")
        answer.append(temp2)
    return answer


print(solution(n, arr1, arr2))

 

해설

or 논리 연산자를 사용한다. 둘 중 하나가 1이면 1이기 때문에 지도를 겹쳤을 때의 결과를 알 수 있다.

해당 결과를 bin을 이용해 2진수로 변환하고, 앞 0의 개수는 n-len(temp1)만큼 추가시킨다.

 

이렇게 얻은 데이터에서 1은 '#'으로, 0은 ' '으로 replace시킨 후, answer 리스트에 추가시킨다.