[이코테] 3-4. 그리디 - 1이 될 때까지

문제

어떠한 수 N이 1이 될 때까지 다음의 두 과정 중 하나를 반복적으로 선책하여 수행하려고 한다. 단, 두 번째 연산은 N이 K로 나누어떨어질 때만 선택할 수 있다.

1. N에서 1을 뺀다.

2. N을 K로 나눈다.

 

예제 입력

입력

17 4

출력

3

 

입력

25 5

출력

2

 

내 풀이

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

count =0

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

print(count)

단순하게 생각해서 1씩 줄여가며 나눠떨어지면 나누고 나누어 떨어지지 않으면 -1을 해주는 식으로 코드를 구현했다.

역시 책에서는 단순하게 푸는 방식이라고 소개되어있다. 비효율적인 방식이란다... 아직 나는 멀었다.

 

책 풀이

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

result =0

while True:
    # (n==k로 나누어떨어지는 수)가 될 때까지 1씩 빼기
    target=(n//k)*k
    result+=(n-target)
    n=target
    # n이 k보다 작을 때(더 이상 나눌 수 없을 때) 반복문 탈출
    if n<k:
        break
    result +=1
    n//=k

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

크게 배울점은

    target=(n//k)*k
    result+=(n-target)
    n=target

이 부분인 것 같다. 나는 한번 연산할 때마다 1번 시행되어 비효율적이었다면

이 코드에서 나눠떨어지는 숫자까지 한번의 연산으로 진행되어 효율적인 코드가 구현된다.