[CodingTest] 1이 될때까지 ( greedy )

◎ 문제

 

어떠한 수 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번의 과정을 수행해야 하는 횟수의 최솟값을 출력한다.

- 입력예시

25 5

- 출력예시 

2

 

 

◎ 문제풀이

 

연산은 2개이다. 

 

1) 1 빼기

2) K로 나누기

 

K는 무조건 2보다 같거나 크므로 연산횟수를 줄이려면 K로 최대한 나누어줘야 한다. (greedy) 

 

◎ 코드

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

while n >= 2 :
  count += 1
  
  if n % k == 0 : n = n/k # 나누기 연산을 먼저 조건 태우기 (greedy)
  else : n = n-1  

  if n == 1 : break

print(count)

 

- 입력

25 5

 

- 출력

2

 

 

 

---

 

 

참고자료

이것이 취업을 위한 코딩 테스트다 with 파이썬 - YES24