[백준/C++] 2839번: 설탕 배달

문제

https://www.acmicpc.net/problem/2839

2839번: 설탕 배달

---

알고리즘 분류

• 수학
• 다이나믹 프로그래밍
• 그리드 알고리즘

 

문제 설명

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

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

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

 

입력

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

 

출력

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

 

소스코드

#include <iostream>
#include <algorithm>
using namespace std;
int dp[5001]; 

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int n;
    cin >> n;
    dp[3] = dp[5] = 1;

    for (int i = 6; i <= n; i++) {
        if (dp[i - 3]) {
            dp[i] = dp[i - 3] + 1;
        }
        if (dp[i - 5]) {
            dp[i] = dp[i] ? min(dp[i], dp[i - 5] + 1) : dp[i - 5] + 1;
        }
    }
    cout << ( dp[n] == 0 ? -1  : dp[n] ) << "\n";
    return 0;
}

i kg에 도달할 때까지 이전에 계산한 (i-3)kg과 (i-5)kg에 필요한 봉지 수에  1씩 더하며 최소 봉지 수를 업데이트해 나가는 방식이다.

global 변수로 dp를 선언해 0으로 초기화된 배열로 선언한 다음, 최소 봉지 수는 1개 이므로 dp[3] = dp[5] = 1; 그 다음 최소 kg은 6부터 시작이므로 6부터 n까지 for loop을 시작한다.

만약 dp[i-3]이 0이 아닐 경우 dp[i]에 dp[i-3]+1로 봉지수를 하나 추가해준다.그리고 dp[i-5]도 0이 아닐 경우엔 앞선 if문에서 dp[i]를 채워줬을 수도 있기 때문에 dp[i]의 값이 존재할 경우 그 값과, dp[i-5]+1을 비교해 더 적은 봉지수를 넣어준다.

만약 dp[i]가 0일 경우, 즉 앞의 조건문이 동작하지 않은 경우엔 dp[i]에 dp[i-5]+1을 넣어준다.

for loop가 끝난 후에 dp[n] == 0이면 -1 출력, 0이 아니라면 dp[n]을 출력한다.