본문 바로가기

두두의 알고리즘/문제

[탐욕법] 난이도1, 2018 E 기업 알고리즘 대회 '1이 될 때까지' (Python)

728x90

<문제>

어떠한 수 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 <= 100000)과 K(2 <= K <= 100000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때, 입력으로 주어지는 N은 항상 K보다 크거나 같다.

<출력 조건>

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

<문제 풀이>

  1. N이 K로 나누어 떨어질 때까지 N에서 1을 빼고, 결괏값에 1을 더한다.
  2. N이 K로 나누어 떨어진다면 N을 K로 나누고 결과값에 1을 더한다.

<코드>

'''입력 예시
17 4

25 5
'''

n,k = map(int,input().split())
count = 0
while n>1:
    while n%k!=0:
        n -= 1
        count += 1
    n /= k
    count += 1
print(count)
#220320
n,k = map(int, input().split())

answer = 0
while n!=1:
    if n%k==0:
        n /= k
        answer += 1
    else:
        n -= 1
        answer += 1
        
print(answer)

 

#이취코 답
n,k = map(int,input().split())
result = 0

while True:
	target = (n//k)*k
	result += (n-target)
	n=target
	if n<k:
		break
	result += 1
	n //= k

result += (n-1)
print(result)

 

<고쳐야 할 점>

  • 반복문 최대한 줄이기