[프로그래머스] 3진법 뒤집기 (Python)

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

코딩테스트 연습 - 3진법 뒤집기

문제 설명

자연수 n이 매개변수로 주어집니다. n을 3진법 상에서 앞뒤로 뒤집은 후, 이를 다시 10진법으로 표현한 수를 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• n은 1 이상 100,000,000 이하인 자연수입니다.

---

입출력 예

n	result
45	7
125	229

---

입출력 예 설명

입출력 예 #1

• 답을 도출하는 과정은 다음과 같습니다.

n (10진법)	n (3진법)	앞뒤 반전(3진법)	10진법으로 표현
45	1200	0021	7

• 따라서 7을 return 해야 합니다.

입출력 예 #2

• 답을 도출하는 과정은 다음과 같습니다.

n (10진법)	n (3진법)	앞뒤 반전(3진법)	10진법으로 표현
125	11122	22111	229

• 따라서 229를 return 해야 합니다.

 

풀이

- 10진법 -> 3진법 : 재귀함수

# 10진수 -> 2~16진수까지 변환
def convert(num, base):
    tmp = '0123456789ABCDEF'
    q, r = divmod(num, base)
    if q == 0:
        return tmp[r]
    else:
        return convert(q) + tmp[r]

- 뒤집기 : arr[::-1]

- n진법 -> 10진법 : int(num, n)

 

코드

# 10진법 -> 3진법
def convert(num):
    tmp = '012'
    q, r = divmod(num, 3)
    if q == 0:
        return tmp[r]
    else:
        return convert(q) + tmp[r]
    
def solution(n):
    # 3진법으로 변환
    ternary = convert(n)
    # 뒤집기
    rev = ternary[::-1]
    # 10진법으로 변환
    return int(rev,3)