[프로그래머스 lv.2] 124 나라의 숫자 (파이썬)

문제

링크

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

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

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

10진법	124나라	10진법	124나라	10진법	124나라
1	1	2	2	3	4
4	11	5	12	6	14
7	21	8	22	9	24
10	41	11	42	12	44
13	111	14	112	15	114
16	121	17	122	18	124

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

제한사항
n은 500,000,000이하의 자연수 입니다.

풀이

1. 재귀로 푸는 문제이다
2. 1, 2, 4가 반복되므로 3으로 나눈 나머지를 index로 하여 '124'에서 맞는 문자를 하나 찾고 정답으로 출력할 문자열 뒤에 붙인다
3. 이때 0이 없는 진수이기 때문에 n-1을 해준다

코드

def solution(n):
    if n < 4:
        return '124'[n-1]
    a, b = divmod(n-1, 3)
    return solution(a) + '124'[b]