[파이썬] 3진법 뒤집기

출처 : 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 해야 합니다.

♡ 나의 풀이

 

def solution(n):
    answer = ''

    while n > 0:			
        n, re=divmod(n,3)	# n을 3으로 나눈 몫과 나머지
        answer+=str(re)
         
    return int(answer, 3)	# 3진법 answer을 10진법으로 변환

 

n을 3으로 나눈 몫이 0보다 클 동안 반복된다. 처음에는 n!=1일 때까지 반복하고 반복문이 끝난 후 answer에 1을 추가하니 시간초과가 떠서 위와 같이 고치니 통과했다.

 

★ 다른 사람 풀이

 

def n_ary(n, base):
    result = []
    while n > 0:
        n, r = divmod(n, base)
        result.append(r)
    return ''.join(map(str, reversed(result)))

def solution(n):
    b3 = n_ary(n, 3)
    b3 = b3[::-1]
    return int(b3, 3)

 

리스트 result를 거꾸로 뒤집은 것을 문자열로 바꿔서 return한다.

b3=b3[::-1] 이렇게 표현하면 뒤집을 수 있다는 것을 배웠다.