파이썬
[이코테] 1이 될 때까지
김고찰
2022. 1. 18. 16:21
시간제한 : 1초
메모리 제한 : 128MB
문제 :
어떠한 수 N이 1이 될 때까지 다음의 두 과정 중 하나를 반복적으로 선택하여 수행하려고 한다. 단, 두 번째 연산은 N이 K로 나누어떨어질 때만 선택할 수 있다.
- N에서 1을 뺀다.
- 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을 빼주고 하는 것이 아닌 하나의 연산식을 통해 횟수를 만들어내는 표현을 배울 수 있었다.