[ 프로그래머스 / PYTHON ] 124 나라의 숫자

문제설명

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

124 나라에는 자연수만 존재합니다.
124 나라에는 모든 수를 표현할 때 1, 2, 4만 사용합니다.
자연수 n이 매개변수로 주어질 때, n을 124 나라에서 사용하는 숫자로 바꾼 값을 return 하도록 solution 함수를 완성해 주세요.

풀이

그냥 10진수를 3진수로 바꾸는 쉬운 문제처럼 보이는 데 살짝 꼬여서 그런지 생각보다 오래 걸렸다.

1. country(n) + num(=[1,2,4] 배열)[나머지-1] 반복
2. 몫이 0 이면 return num[나머지-1] -> 재귀 종결 조건 기술

위 과정에서 3의 배수와 3을 처리해주는 elif문만 넣으면 된다.

코드

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

    def country(n):
        q,r = divmod(n,3)
        if q==0: return num[r-1]
        elif q==1 and r==0: return num[2]
        elif r==0:  return country(q-1) + num[2]
        else:   return country(q) + num[r-1]

    return country(n)