1이 될 때까지 [그리디 알고리즘]

그리디 알고리즘 실전 문제

1이 될 때까지

🐾 문제 설명

어떠한 수 N이 1이 될 때까지 다음의 두 과정 중 하나를 반복적으로 선택하여 수행하려고 한다. 단, 두 번째 연산은 N이 K로 나누어떨어질 때만 선택할 수 있다.
1. N에서 1을 뺀다.
2. N을 K로 나눈다.
N과 K가 주어질 때 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 작성하시오.

🐾 입출력 예시

• 입력
  25 5
• 출력
  2

🐾 내 풀이

N, K = map(int, input().split())

cnt = 0
while N != 1:
    if N % K == 0:
        N = N // K
    else:
        N -= 1
    cnt += 1

print(cnt)

🐾 답안 예시

< 이것이 취업을 위한 코딩 테스트다 with 파이썬 > 답안 예시

n, k = map(int, input().split())
result = 0

while True:
    target = (n // k) * k
    result += (n - target)
    n = target
    if n < k:
        break
    result += 1
    n //= k

result += (n - 1)
print(result)

어떤 수가 있을때, 이 수를 '2 이상의 수로 나누는 것'이 항상 '1을 빼는 것'보다 해당 수를 더 많이 작게 만들 수 있으므로 최대한 많이 나눌 수 있도록 코드를 작성한 것이 책의 답안 예시이다.

참고자료
이것이 취업을 위한 코딩 테스트다 with 파이썬 - 나동빈