파이썬

[이코테] 1이 될 때까지

김고찰 2022. 1. 18. 16:21

시간제한 : 1초 

메모리 제한 : 128MB


문제 :

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

  1. N에서 1을 뺀다.
  2. N을 K로 나눈다.

입력 조건 :

첫째 줄에 N(2<=N<=100,000)과 K(2<=K<=100,000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때 입력으로 주어지는 N은 항상 K보다 크거나 같다.


출력 조건 :

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

 


입력 예시 : 

25 5

출력 예시 : 

2


나의 풀이 :

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

이렇게 푼다고 해서 지금 조건에서는 문제가 생기진 않는다. 하지만 만약 N이 더욱 큰 수의 범위를 갖게 되는 경우에도 코드가 빠르게 동작하기 위해서는 N이 K의 배수가 되도록 효율적으로 한 번에 만들어주는 코드를 작성해야 한다.

 

답안 예시 : 

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)

(n//k)*k 표현이 다소 생소하다. 

-1 연산을 반복해주는 것이 아닌 한번의 연산으로 K로 N이 나누어떨어질 수 있도록 만들어주는 연산이므로 코드의 속도가 훨씬 빨라질 수 있다. 반복문을 돌면서 -1을 빼주고 하는 것이 아닌 하나의 연산식을 통해 횟수를 만들어내는 표현을 배울 수 있었다.