[Python] 이코테_1이 될 때까지

💡 1이 될 때까지
이것이 취업을 위한 코딩 테스트다 with 파이썬 (나동빈)
: 99p

---

문제

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

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

• 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

---

풀이

아이디어

• 그리디 알고리즘
• N이 K로 나누어떨어지면 N에서 1을 뺄 때보다 연산 과정을 줄일 수 있음

코드

• 내가 작성한 코드

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

count = 0

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

print(count)

💡 하지만 N이 100억 이상의 큰 수라면?

• 답안 예시

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
    # K로 나누기
    result += 1
    n //= k

# 마지막으로 남은 수에 대하여 1씩 빼기
result += (n - 1)
print(result)

제발 주어진 케이스만 생각하지 말고, 큰 수까지 넓게 넓게 생각해보자..