코딩 테스트 연습/[프로그래머스] Python

[LV.2] 월간 코드 챌린지 시즌1 > 3진법 뒤집기

duswjd_data 2025. 6. 27. 09:07

문제

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

 

프로그래머스

SW개발자를 위한 평가, 교육의 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr


문제 설명

  • 자연수 n이 매개변수로 주어질 때,
    n을 3진법 상에서 앞뒤로 뒤집은 후, 이를 다시 10진법으로 표현한 수를 반환
  • 제한사항) n은 1 이상 100,000,000 이하인 자연수


정답 코드

def solution(n):
    step1 = ''
    while n > 0:
        step1 = str(n % 3) + step1
        n //= 3
    step2 = step1[::-1]
    return int(step2, 3)

코드 설명

  • step1 = ''
    • 3진법 문자열을 만들기 위한 초기화
  • str(n % 3)
    • n을 3으로 나눈 나머지(3진법의 한 자리)를 문자열로 변환하여 붙이기
    • str(n % 3) + step1
      • 자릿수 순서를 맞추기 위해 앞쪽에 붙임 (뒤에 붙이면 거꾸로 됨)
  • n //= 3
    • n을 3으로 나눠서 몫으로 갱신
  • step2 = step1[::-1]
    • 3진수 문자열 뒤집기
  • int(step2, 3)
    • 뒤집은 3진수를 다시 10진수로 해석
    • int(문자열, 진법) 함수 → 진법을 10진수로 바꿔줌

개선된 코드 (divmod() 사용)

def solution(n):
    base3 = ''
    while n > 0:
        n, r = divmod(n, 3)
        base3 = str(r) + base3
    return int(base3[::-1], 3)

코드 설명

  • divmod(n, 3)
    • n을 3으로 나눈 몫(n)과 나머지(r)를 동시에 반환 (더 효율적)
  • base3 = str(r) + base3
    • (나머지를 문자열로 바꾼) 새로운 자릿수를 앞에 붙임
  • base3[::-1]
    • 3진법 문자열을 뒤집음
  • int(문자열, 3)
    • 문자열을 3진법 수로 보고 10진법으로 변환

Q) str(r) 쓰는 이유

  • base3는 문자열인데 r은 정수이기 때문에, 문자열끼리 이어붙이기 위해 str()로 변환