프로그래머스 124 나라의 숫자 (파이썬)

구현, Leve 2

 

문제: 124 나라의 숫자 

코딩테스트 연습 - 124 나라의 숫자

문제 설명

124 나라가 있습니다. 124 나라에서는 10진법이 아닌 다음과 같은 자신들만의 규칙으로 수를 표현합니다.

1. 124 나라에는 자연수만 존재합니다.
2. 124 나라에는 모든 수를 표현할 때 1, 2, 4만 사용합니다.

예를 들어서 124 나라에서 사용하는 숫자는 다음과 같이 변환됩니다.

10진법	124 나라	10진법	124 나라
1	1	6	14
2	2	7	21
3	4	8	22
4	11	9	24
5	12	10	41

자연수 n이 매개변수로 주어질 때, n을 124 나라에서 사용하는 숫자로 바꾼 값을 return 하도록 solution 함수를 완성해 주세요.

제한사항

• n은 500,000,000이하의 자연수 입니다.

입출력 예

n	result
1	1
2	2
3	4
4	11

첫번째 제출 코드 (정답)

def solution(n):
    answer = ''
    if n < 3:
        return str(n)
    if n == 3:
        return "4"
    
    while True:
        b = n%3
        if b == 0:
            answer = '4' + answer
            n = n//3 - 1
            if n <= 0:
                break
        
        else:
            n = n//3
            if b == 3:
                b = 4
            answer = str(b) + answer
            if n == 0:
                break
            
        if  n < 3:
            answer = str(n) + answer
            break
            
    return answer

 

결과 및 정리

일단 풀었는데 너무 비효율적으로 짠 것 같습니다.

처음에는 10진수를 3진수로 변환하는 것과 비슷해보여 쉽게 접근했다가 1부터 시작하는 것과 n을 나눌 때 나머지가 0이면 4가 뒤에 붙고 n은 1 빼주어야 하는 규칙이 발견되어 위와 같이 풀었습니다.

효율적이고 직관적인 다른 분의 코드

def solution(n):
    answer = ''
    num = ['1','2','4']

    while n > 0:
        n -= 1
        answer = num[n % 3] + answer
        n //= 3

    return answer

 

프로그래머스에서 좋아요를 가장 많이 받은 코드입니다.

먼저 배열 안에 1,2,4만 넣고 인덱스 기준이므로 n-1을 해준 것이 핵심인 것 같습니다.