[python] 3진법 뒤집기 (strip)

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

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

코드 구현 

def solution(n):
    answer = 0
    num = ''
    while n > 0:
        num = str(n % 3) + num
        n //= 3
    num = num.strip('0')
    i = 0
    while i < len(num):
        answer += int(num[i])*(3**i)
        i += 1
    return answer

 

틀린 풀이

def solution(n):
    answer = 0
    num = ''
    while n > 0:
        num = str(n % 3) + num if n % 3 != 0 else num
        n //= 3
    i = 0
    while i < len(num):
        answer += int(num[i])*(3**i)
        i += 1
    return answer

 test case 1에서 숫자 45를 3진법으로 변환할 때, '1200'을 '12'로 만들기 위해서 삼항 연산자를 추가하였다.

하지만 저렇게 되면 3진법으로 변환한 숫자 중간에 0이 포함된다면 이마저 제외되어 버리고 만다.

ex) 11 -> 3진법 시 102(3) -> 그러나 102(3)이 아닌 12(3)으로 저장됨 

따라서 가장자리 0을 제외하고 싶다면 strip 함수를 이용하면 된다.