[알고리즘] 백준 2839번 문제

date

Nov 5, 2023

slug

dp-2

author

status

Public

2839번 문제

문제

상근이는 요즘 설탕공장에서 설탕을 배달하고 있다. 상근이는 지금 사탕가게에 설탕을 정확하게 N킬로그램을 배달해야 한다. 설탕공장에서 만드는 설탕은 봉지에 담겨져 있다. 봉지는 3킬로그램 봉지와 5킬로그램 봉지가 있다.

상근이는 귀찮기 때문에, 최대한 적은 봉지를 들고 가려고 한다. 예를 들어, 18킬로그램 설탕을 배달해야 할 때, 3킬로그램 봉지 6개를 가져가도 되지만, 5킬로그램 3개와 3킬로그램 1개를 배달하면, 더 적은 개수의 봉지를 배달할 수 있다.

상근이가 설탕을 정확하게 N킬로그램 배달해야 할 때, 봉지 몇 개를 가져가면 되는지 그 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 N이 주어진다. (3 ≤ N ≤ 5000)

출력

상근이가 배달하는 봉지의 최소 개수를 출력한다. 만약, 정확하게 N킬로그램을 만들 수 없다면 -1을 출력한다.

예제 입력 1

예제 출력 1

코드


import sys
input = sys.stdin.readline
n = int(input())
dp = [-1]*5001
dp[3]=1
dp[5]=1
for i in range(6,n+1):
    if i%5 == 0:
        dp[i]=dp[i-5]+1
    elif i%3 == 0:
        dp[i]=dp[i-3]+1
    elif dp[i-3]>0 and dp[i-5]>0:
        dp[i]=min(dp[i-3], dp[i-5])+1
print(dp[n])

풀이

3키로, 5키로인 경우에는 1개라는 답이 바로 나오므로 dp 필요 X

따라서 6부터 상향식으로 반복 시작함.

가질 수 있는 경우의 수는 총 3가지다.

5키로로만 나눌 수 있는 경우

3키로로만 나눌 수 있는 경우

3키로와 5키로로 둘 다 나눌 수 있는 경우

따라서 3가지 경우에 해당하도록 if문으로 나눠주면 된다.

만약, 5키로로만 나눌 수 있는 경우라면

5키로로 나눌 수 있는 이전 수의 경우에 한 봉지씩만 추가하면 됨. 따라서 dp[i]=dp[i-5]+1

만약, 3키로로만 나눌 수 있는 경우라면

3키로로 나눌 수 있는 이전 수의 경우에 한 봉지씩만 추가하면 됨. 따라서 dp[i]=dp[i-3]+1

그러나, 3키로와 5키로 모두 이용해서만 나눌 수 있는 경우라면 두 경우의 이전 경우를 확인해서 더 작은 경우를 선택, 한 봉지를 추가한다. dp[i]=min(dp[i-3], dp[i-5])+1

[알고리즘] 백준 2839번 문제 - DP

2839번 문제

문제

입력

출력

예제 입력 1

예제 출력 1

코드

풀이