[ 이것이 취업을 위한 코딩 테스트다 with 파이썬 ] Chapter 03 그리디 - 실전문제: 1이 될 때까지

1이 될 때까지

 

1 _ 문제

어떠한 수 N이 될 때까지 다음의 두 과정 중 하나를 반복적으로 선택하여 수행하려고 한다. 단, 두 번째 연산은 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 ≤ 100,000) 과 K(2 ≤ K ≤ 100,000)가 공백으로 구분되며 각각 자연수로 주어진다. 이때 압력으로 주어지는 N은 항상 K보다 크거나 같다.
• 출력 조건
  • 첫째 줄에 N이 1이 될 때까지 1번 혹은 2번의 과정을 수행해야 하는 횟수의 최솟값을 출력한다.
• 입출력 예시

 

---

 

2 _ 문제 해결(고민) 과정

N을 K로 나눈 다음, 나머지 만큼 1번의 과정을 반복해서 N을 K의 배수로 만든다.
그 다음에, N을 K로 나누면서 결과적으로 N을 1로 만든다.
이러한 과정에서 계산된 횟수를 구하면 그게 최소 횟수가 된다고 생각한다.

 

---

 

3 _ 문제 알고리즘

1. 횟수 변수 count 선언

2. N = 1이 될때 까지 다음의 과정을 반복한다.

   2 - 1. N을 K로 나눈 나머지를 구한다.

   2 - 2. 구한 나머지를 count 에 더하고, N = N - (나머지) 할당

   2 - 3. N 을 K로 나눈다.

   2 - 4. count 에 1을 더한다.

3. count 값을 출력한다.

 

---

 

4 _ MY CODE

• 코드

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

count = 0

while(n != 1):
    r = n%k
    count += r
    n -= r
    n /= k
    count += 1

print(int(count))

• 실행 결과

 

---

 

5 _ 교재 답안 코드

• 단순하게 푸는 답안 예시

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

while n >=k:
    while n % k != 0:
        n -= 1
        result += 1
    n //= k
    result += 1

while n > 1:
    n -= 1
    result += 1

print(result)

• 답안 예시

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)

 

---

 

6 _ 마치며

얼른 씻고 자야지.

 

---

 

위 내용은 저자 나동빈의 < 이것이 취업을 위한 코딩 테스트다 with 파이썬 > 을 읽고, 공부한 내용입니다.

https://book.naver.com/bookdb/book_detail.nhn?bid=16439154

이것이 취업을 위한 코딩 테스트다 with 파이썬