[프로그래머스] 124 나라의 숫자(Python)

https://programmers.co.kr/learn/courses/30/lessons/12899

코딩테스트 연습 - 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

풀이 설명

1. 규칙을 찾는다!

 -> 10진수를 124나라의 숫자로 변환하려면,

     arr[n % 3]을 빈 문자열의 앞에 추가한다.

     추가한 후 n은 n//3으로 바꿔주고, n이 0이 될 때까지 반복

Python3 코드

def solution(n):
    arr = [4,1,2]
    
    if n <= 3:
        return str(arr[n % 3])
    else:
        answer = ''
        add = ''
        while n > 0:
            add = str(arr[n % 3])
            answer = add + answer
            if n % 3 == 0:
                n = n // 3 - 1
            else:
                n = n // 3
        return answer