[프로그래머스] level2. 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 함수를 완성해 주세요.

 

나의 풀이

def solution(n):
    answer = ''
    while n :
        num = n % 3 
        if num == 1:
            ans = '1'
        elif num == 2:
            ans = '2'
        else:
            ans = '4'
            n -= 1
        answer = ans + answer
        n = n // 3
    return answer

위와 같은 코드를 좀 더 간결하게 다음과 같이 바꿀 수도 있습니다.

def solution(n):
    answer = ''
    num124 = ['1','2','4']
    while n:
        n -= 1
        answer = num124[n % 3] + answer
        n = n // 3
    return answer

 

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12899

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