[python] 34. 124 나라의 숫자 ★

문제 출처

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

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

 

문제 설명

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

  - 124 나라에는 자연수만 존재합니다.

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

 

---

나의 코드

def solution(n):
    num = ['1', '2', '4']
    answer = ''
    
    while True:
        if n <= 3:
            answer += num[n-1]
            return ''.join(list(reversed(answer)))  #answer이 역방향으로 숫자가 이어붙여졌기 때문에 출력할 때는 반대로 출력한다.
        answer += num[(n-1)%3]  #1)
        n = (n-1)//3  #2)

이거 푸는데 한시간이 넘게 걸렸다ㅠㅠ 나는 바보다ㅠㅠ

 

1, 2, 3/ 4, 5, 6/ 7, 8, 9 이런 식으로 124나라 표현식에 패턴이 보이는 것을 보고, 각 수를 3으로 나눈 몫과 나머지를 이용해야겠다는 생각을 했다.

1) n-1의 나머지를 구해서 그 값이 0이면 '1', 1이면 '2', 2이면 '4'가 출력되게 설정했다.

2) n-1의 몫을 n으로 재설정해서 다음 whlie문을 수행하는 방식이다.

3) 1번과 2번을 반복하다가 n이 3이하가 된다면 마지막으로 num[n-1]를 수행해서 최종 answer를 출력한다.

  (단, '+='를 수행하다보니 answer이 역방향으로 생성되어서 출력할 때는 다시 역방향으로 출력되게 작성하기 위해 reversed함수를 사용했다.)

 

 

다른 풀이

def solution(n):
    if n<=3:
        return '124'[n-1]
    else:
        q, r = divmod(n-1, 3) 
        return solution(q) + '124'[r]

이렇게 재귀로도 풀 수 있다. 어쩜,,,, 이 문제는 껌이지라고 말하듯이 간결하고 앀크하게 풀 수 있을까. 나는 아직도 멀었다.