[이것이 코딩 테스트다 with Python] 그리디 : 1이 될 때까지

교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 3 그리디
실전문제 3-4 1이 될 때까지 99p

 

문제

어떠한 수 N이 1이 될 때 까지 다음의 두 과정 중 하나를 반복적으로 선택해 수행하려 합니다.

단, 두 번째 연산은 N이 K로 나누어떨어질 때만 선택할 수 있습니다.

1.  N에서 1을 뺀다.
2.  N을 K로 나눈다.

예를 들어 N이 17, K가 4라고 가정하겠습니다. 이때 1번의 과정을 한 번 수행하면 N은 16이 됩니다.

이후에 2번의 과정을 두 번 수행하면 N은 1이 됩니다. 결과적으로 이 경우 전체 과정을 실행한 횟수는 3이 됩니다.

 

이는 N을 1로 만드는 최소 횟수라 할 수 있습니다.

N과 K가 주어질 때 N이 1이 될 때 까지 1번 혹은 2번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 만들어보세요.

 

 

입력 조건

• 첫째 줄에 N(2 ≤ N ≤ 100,000)과 K(2 ≤ K ≤ 100,000)가 가 공백으로 구분되며 각각 자연수로 주어진다.
• 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.

 

출력 조건

• 첫째 줄에 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 횟수의 최솟값을 출력한다.

 

 

---

✔ Solution

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

while True:
    if N == 1 :  # N이 1일때 수행을 멈춤
        break
    if N % K == 0 :
        N /= K
        cnt += 1
    else :
        N -= 1
        cnt += 1
        
print(cnt)

 

 

✔ Solution 2

# N, K공백을 기준으로 구분하여 입력 받기
n, k = map(int, input().split())

result = 0

while True:
    # N이 K로 나누어 떨어지는 수가 될 때까지만 1씩 빼기
    target = (n // k) * k
    result += (n - target)
    n = target
    # N이 K보다 작을 때 (더 이상 나눌 수 없을 때) 반복문 탈출
    if n < k:
        break
    # K로 나누기
    result += 1
    n //= k

# 마지막으로 남은 수에 대하여 1씩 빼기
result += (n - 1)
print(result)