[이것이 코딩 테스트다 with 파이썬] 그리디 - 1이 될 때까지

문제

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

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

N에서 1을 뺀다.

N을 K로 나눈다.

N이 1이 될 때 까지 1번 혹은 2번의 과정을 수행해야하는 최소 횟수를 구하는 프로그램을 작성

입력 조건

첫째 줄에 N(2<= N <= 100000)과 K(2 <= K <= 100000)가 공백으로 구분되며 각각 자연수로 주어짐

이때, 입력으로 주어지는 N은 항상 K보다 크거나 같음

출력 조건

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

입력 예시
3 3
3 1 2
4 1 4
2 2 2

출력 예시
2

문제 해설

• 최대한 많이 나누기

풀이

나의 풀이

import sys

n,k = map(int,sys.stdin.readline().split())
count = 0

while n != 1:
  if n % k != 0:
    n = n -1
    count += 1
  else:
    n = n / k
    count += 1

print(count)

다른 풀이 (n이 100억 이상일때를 생각하라)

import sys
n, k = map(int,sys.stdin.readline().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)