[파이썬] 124 나라의 숫자

출처 : 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

 

♡ 나의 풀이

 

answer = ''
def solution(n):
    if n<=3:			# n이 3이하면 '124'문자열에서 [n-1]반환. 종료조건
        return '124'[n-1]
    answer, r =(n-1)//3, (n-1)%3
    return solution(answer)+'124'[r]	# 재귀

 

1='124'[1-1], 2='124'[2-1], 3='124[3-1]로 보아서 n이 3이하일 때 '124'[n-1]인 것을 알 수 있다.

그 이상부터는 4=(4-1)//3 + '124'[(3-1)%4]이다.

10=(10-1)//3+'124'[(10-1)%3]

='124'[3-1] + '124'[0]

=41

이 문제의 규칙를 알고, 재귀를 사용할 줄 알면 쉽게 풀 수 있는 문제다.

 

☆ 다른 사람 풀이

 

def change124(n):
    num = ['1','2','4']
    answer = ""
    while n > 0:
        n -= 1
        answer = num[n % 3] + answer
        n //= 3
    return answer

 

기발하게 잘 짠 코드라는 생각이 든다.

 

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

 

divmod를 이용한 풀이.

q에 몫, r은 나머지로 초기화.