[ 이것이 코딩테스트다 ] 1이 될 때까지 - 그리디 알고리즘

[  이것이 취업을 위한 코딩테스트다 with 파이썬  ] 1이 될 때까지 _ Python 파이썬

 

" 문제 "

 

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

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

 

1. N에서 1을 뺀다.

2. N을 K로 나눈다.

 

N과 K가 주어질 때 N이 1이 될 때까지

1번 혹은 2번의 과정을 수행해야 하는 최소 횟수를 구하는 프로그램을 작성하시오.

 

<입력 조건>

첫재 쭐에 N (2 <= N <= 100,000) 과 K (2 <= K <= 100,000) 가 공백으로 구분되며

각각 자연수로 주어진다. 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.

 

<출력 조건>

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

---

" 아이디어 "

연산을 가장 적게 사용하면서 주어진 수를 1로 만드려면

가장 많은 횟수의 나눗셈을 해야한다.

---

" 코드 "

n, k = map(int, input().split())
cnt = 0
while n > k:
  if n % k == 0:
    n //= k
    cnt += 1
  else:
    mod = n % k
    n -= mod
    cnt += mod
  
  if n == 1:
    break
 
if n % k == 0:
  cnt += 1
else:
  cnt += (n-1)
  
print(cnt)

 

---

" 참고자료 "