[1차] 비밀지도 / 프로그래머스, 파이썬 코딩테스트 연습2018 KAKAO BLIND RECRUITMENT

문제

네오는 평소 프로도가 비상금을 숨겨놓는 장소를 알려줄 비밀지도를 손에 넣었다. 그런데 이 비밀지도는 숫자로 암호화되어 있어 위치를 확인하기 위해서는 암호를 해독해야 한다. 다행히 지도 암호를 해독할 방법을 적어놓은 메모도 함께 발견했다.
지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자.

1. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다.
2. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 1, 공백 부분을 0으로 부호화했을 때 얻어지는 이진수에 해당하는 값의 배열이다.

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

 

 

제출답안

def solution(n, arr1, arr2):
    answer = []
    def d2b_r(num):
        """Decimal to Binary in Reverse"""
        output, i = [], 0
        while(num):
            num, remainder = divmod(num, 2)
            output.append(remainder)
            i+=1
        if n-i:
            output.extend([0]*(n-i))
        
        return output
        
    arr1, arr2 = map(d2b_r, arr1), map(d2b_r, arr2)
    for ar1, ar2 in zip(arr1, arr2):
        result = ['#' if val >0 else ' ' for val in map(lambda x, y: x+y, ar1, ar2)]
        answer.append(''.join(result[::-1]))
    
    return answer

 

제출 후 개선답안

 

개선내용
rjust함수 사용

가독성 향상

def solution(n, arr1, arr2):
    answer = []
    def d2n(num):
        output = ''
        while(num):
            num, remainder = divmod(num, 2)
            output += '#' if remainder else ' '
        return output[::-1].rjust(n, ' ')
        
    for i,j in zip(arr1,arr2):
        row = ''.join([' ' if a1+a2=='  ' else '#' for a1, a2 in zip(d2n(i), d2n(j))])
        answer.append(row)
    return answer

 

참고코드

 

 

개선 전수 성능 비교

문제 바로가기