파이썬 문제: 4/5

프로그래머스 Level2 : 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

해설

처음에 3진법과 124 숫자랑 매칭되는 규칙이 너무 안떠올라서 삽질을 진짜 많이했다. 정확성 검사를 계속 실패했는데, 한번 성공하니까 효율성까지 같이 통과되더라

def solution(n):
    answer = []
    while n != 0:  #3진법으로 변환
        answer.append(n%3)
        n = n//3   
    for i in range(len(answer)-1): #3진법 -> 124
        if answer[i] <= 0:
            answer[i+1] -= 1
    if answer[-1] == 0:
        answer.pop()
    return ''.join(map(str, reversed(answer))).replace('0', '4').replace('-1', '2')

애초에 3진법으로 만들지 않고 바로 124로 매칭 시키는 방법이 있었다. 훨씬 간단하고 직관적이다.

def solution(n):
    if n<=3:
        return '124'[n-1]
    else:
        q, r = divmod(n-1, 3) #몫/나머지
        return solution(q) + '124'[r]