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

▶ 문제 설명

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

1. 지도는 한 변의 길이가 n인 정사각형 배열 형태로, 각 칸은 "공백"(" ") 또는 "벽"("#") 두 종류로 이루어져 있다.
2. 전체 지도는 두 장의 지도를 겹쳐서 얻을 수 있다. 각각 "지도 1"과 "지도 2"라고 하자. 지도 1 또는 지도 2 중 어느 하나라도 벽인 부분은 전체 지도에서도 벽이다. 지도 1과 지도 2에서 모두 공백인 부분은 전체 지도에서도 공백이다.
3. "지도 1"과 "지도 2"는 각각 정수 배열로 암호화되어 있다.
4. 암호화된 배열은 지도의 각 가로줄에서 벽 부분을 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]
출력	["######", "### #", "## ##", " #### ", " #####", "### # "]

▶ 내가 짠 코드

def solution(n, arr1, arr2):
    answer = []
    a1=[]
    a2=[]
    for i in range(n):
        x = bin(arr1[i])[2:]
        y = bin(arr2[i])[2:]
        a1.append(x.zfill(n))
        a2.append(y.zfill(n))
    for j in range(n):
        val = bin(int(a1[j],2) | int(a2[j],2))[2:]
        s=''
        for k in val:
            if k == '0':
                s += ' '
            else:
                s += '#'
        answer.append(s)
    return answer

비트연산 문제라는 생각이 들었는데 복병이 몇가지 있었다.

자리수에 맞게 0을 채워넣어야 하는 것과

비트연산 후 각각 문자를 판단하여 #과 공백으로 치환해야 하는 것.

전달받은 값을 이진수 문자열로 돌려주는 bin()함수를 사용했다. 이때 앞에 0b가 붙으므로 제거해주기 위해 슬라이싱을 했다.

그리고 새로운 배열에 각각 이진수로 변환한 값을 append하면서 zfill()을 이용해 자릿수만큼 앞자리가 채워져있지 않으면 0을 채워넣도록 하였다.

그리고 다시 for문을 돌면서 OR 연산을 하였고 문자열 각각의 자릿수를 판단하였다.

 

이렇게 했더니 두번째 테스트 케이스를 통과하지 못했는데 

분석해보니 zfill()을 사용하여 append한 배열 각각의 원소를 OR연산을 해줄 경우 앞자리가 0이면 다시 자릿수가 맞지 않는 문제가 발생하기 때문이었다.

수정 후

def solution(n, arr1, arr2):
    answer = []
    a1=[]
    a2=[]
    for i in range(n):
        x = bin(arr1[i])[2:]
        y = bin(arr2[i])[2:]
        a1.append(x.zfill(n))
        a2.append(y.rjust(n,'0'))
    for j in range(n):
        val = bin(int(a1[j],2) | int(a2[j],2))[2:].zfill(n)
        s=''
        for k in val:
            if k == '0':
                s += ' '
            else:
                s += '#'
        answer.append(s)
    return answer

따라서 OR연산 후 다시 zfill()을 해줌으로써 문제를 해결할 수 있었다.

▶ 참고사항

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

for문을 두번이나 돌린게 효율성 측면에서 안좋다고 생각했는데

다른사람의 풀이를 참고해보니 역시나였다.

굳이 새로운 배열에 append할 필요 없이 바로 인덱스별로 OR연산 후

rjust를 사용하여 자릿수만큼 0을 채워주었고

그 이후 문자열을 치환했다.

필요없는 과정이나 생략할 수 있는 부분들을 최대한 생략하여 효율적인 코드를 짜는 것에 주의해야겠다.