새소식

🧇 Algorithm/백준

[백준] 14916 - 거스름돈

  • -
728x90

문제 ( www.acmicpc.net/problem/14916 )

춘향이는 편의점 카운터에서 일한다.

손님이 2원짜리와 5원짜리로만 거스름돈을 달라고 한다. 2원짜리 동전과 5원짜리 동전은 무한정 많이 가지고 있다. 동전의 개수가 최소가 되도록 거슬러 주어야 한다. 거스름돈이 n인 경우, 최소 동전의 개수가 몇 개인지 알려주는 프로그램을 작성하시오.

예를 들어, 거스름돈이 15원이면 5원짜리 3개를, 거스름돈이 14원이면 5원짜리 2개와 2원짜리 2개로 총 4개를, 거스름돈이 13원이면 5원짜리 1개와 2원짜리 4개로 총 5개를 주어야 동전의 개수가 최소가 된다.

 

입력

첫째 줄에 거스름돈 액수 n(1 ≤ n ≤ 100,000)이 주어진다.

 

출력

거스름돈 동전의 최소 개수를 출력한다. 만약 거슬러 줄 수 없으면 -1을 출력한다.

 

예제

13

5

 

14

4

 

나의 코드

import sys
sys.setrecursionlimit(1000000)


def dp(n):
    if n == 0: return mem[n]
    if n < 0: return 100001  # return max

    res = mem[n]
    if res != -1:  # already calculated
        return res

    mem[n] = 100001  # initiate as max
    res = min(mem[n], dp(n - 5) + 1)  # coin 5
    res = min(res, dp(n - 2) + 1)  # coin 2

    return res


n = int(input())
mem = [-1 for i in range(100001)]  # memoization
result = dp(n)
if result > 100000:  # can't change
    print(-1)
else:
    print(result + 1)
  • memoization 구현 → 이미 계산된 값을 return하여 시간 단축!
  • mem배열 크기를 n으로 해도 될 듯
  • min함수를 사용하여 100001, 5원, 2원의 경우 중 가장 작은 값을 return하여 최소값을 찾다.
  • 백준이 파이썬 재귀 횟수를 제한하므로 이를 풀어주기 위해 가장 위 2줄 코드를 추가

 

💫 이 방법이 아니라 그냥 반복문으로 하면 훨씬 간단하게 해결 가능

728x90
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.