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

문제 설명

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이하의 자연수 입니다.

---

입출력 예

nresult

1	1
2	2
3	4
4	11

나의 풀이

패턴을 코드로 옮겨보는 데서 뭔가 지독하게 꼬여서 잘 안 풀린 느낌이 있습니다. 3진법에서 1씩 빠진 느낌인 건 이해를 했는데, 이걸 단순하게 코드로 적기가 힘들어서 재귀와 분기까지 두면서 풀었는데, 제출하고 참고해보니 더 간단하게 해결하는 방법이 있었습니다. 

초기 풀이

def dfs(n, nums):
    if (n < 4):
        return str(nums[n % 3 - 1])
    if (n % 3 == 0):
        n = n - 3
    return dfs(n // 3, nums) + dfs(n % 3, nums)        

def solution(n):
    nums = [1, 2, 4]
    return dfs(n, nums)

개선된 풀이

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

이런식으로 반복문을 진입할 때마다 1씩 빼주면서 하면 패턴에 맞게 해결할 수 있습니다.