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

이 글은 이것이 취업을 위한 코딩테스트다 책을 기반으로 쓰여진 글입니다.
출처 : 이것이 취업을 위한 코딩테스트다

---

문제 

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

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

 

1. N에서 1을 뺀다.

2. N을 K로 나눈다.

 

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

 

입력 조건

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

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

 

출력 조건

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

---

알고리즘

1. N을 K로 나눈 나머지가 0일 경우 N을 K로 나누고 몫을 다시 N으로 저장

2. answer 1 증가

2. N을 K로 나눈 나머지가 0이 아닐 경우 N에서 1을 빼고 N으로 저장

3. answer 1 증가

4. N이 1이 되면 종료하고 answer 출력

 

* 나눌 수 있으면 우선 나누고 아니면 -1을 빼는게 중요 

---> 그래야 최소한의 횟수가 나옴

 

내 소스 코드

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

answer = 0

while N != 1:
    if N % K == 0 :
        N = N // K
        answer += 1

    else :
        N -= 1
        answer += 1

print(answer)

 

결과