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

Link

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

 

Solution

정말 Other Solution의 풀이를 보고 감탄을 할 수 밖에 없었다.

시간복잡도도 시간복잡도지만 bin(i | j) 연산을 보고 매우 놀랐다. 해당 연산은 i와 j를 이진수로 바꾼 다음 or 연산을 사용한다.

이 문제를 통해 rjust(), zfill(), bin(i | j)를 익히고 가길 바란다.

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

    arr1 = ['0' * (n - len(bin(i)[2:])) + bin(i)[2:] for i in arr1]
    arr2 = ['0' * (n - len(bin(i)[2:])) + bin(i)[2:] for i in arr2]

    for i in range(n):
        t = ''
        for j in range(n):
            if arr1[i][j] == '0' and arr2[i][j] == '0':
                t += '0'
            else:
                t += '1'
        answer.append(t)

    for i in range(n):
        answer[i] = answer[i].replace('1', '#')
        answer[i] = answer[i].replace('0', ' ')

    return answer

🔑 key point: replace()

 

Other Solution

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

    for i, j in zip(arr1, arr2):
        t = str(bin(i | j)[2:])
        t = t.rjust(n, '0')
        t = t.replace('1', '#')
        t = t.replace('0', ' ')
        answer.append(t)

    return answer

🔑 key point: zip(), bin(i | j), rjust()(zfill()과 유사한 기능)