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을 출력한다.

---

풀이

DP와 2차원 배열을 이용해서 풀었다. 

dp[0][num] = 3kg만 가지고 있는데 num kg를 줘야하는 경우 num이 3의 배수면 3으로 나눈 몫을 저장하고, 3으로 나눠지지 않는 것은 -1를 넣었다.

 

dp[1][num] = 3,5 kg 가지고 있는 경우  

dp[1][num]은

1. 5로만 나눠지는 경우에는 5로 나눈 몫을 저장
2. 5로만 나눠지지 않을 경우 3가지고만 나눠지면 dp[0][num] 값을 저장
3. 1,2번이 안될 때 5를 뺀 dp[1][num-5]가 -1이 아니면 dp[1][num]  = dp[1][num-5]+1
4. 1,2,3번이 다 안될 경우 -1

ex) 18

1. 5로만 나눠지지 않으니 x

2. 3로만 나눠지지 않으니 x

3 dp[1][13]은 (5,5,3) 3이니 여기에 더하기 1해서 dp[1][18] =4 

 

ex) 19

1. 5로만 나눠지지 않으니 x

2. 3로만 나눠지지 않으니 x

3. dp[1][14]는 -1 x

4. dp[1][19] = -1

---

#include <bits/stdc++.h>
using namespace std;
int dp[2][5001] = { 0, }; 
// dp[0][num] = 3kg만 가지고 있을 경우
// dp[1][num] = 3kg, 5kg 가지고 있을 경우
int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	int num;
	cin >> num;
	dp[0][1] = -1;
	dp[0][2] = -1;
	dp[0][3] = 1;
	dp[1][1] = -1;
	dp[1][2] = -1;
	dp[1][3] = 1;
	dp[1][4] = -1;
	dp[1][5] = 1;
	for (int i = 4; i <= 5000; i++) { 
		if (i % 3 == 0) {
			dp[0][i] = i / 3;
		}
		else {
			dp[0][i] = -1;
		}
	}
	
	for (int i = 6; i <= 5000; i++) {
		if (dp[1][i - 5] == -1) { 
			if (i % 5 == 0) { // 5원으로만 가능한 경우
				dp[1][i] = i/5;
			}
			else { 
				if (dp[0][i] != -1) { //3원가지고만 가능한 경우
					dp[1][i] = dp[0][i];
				}
				else {
					dp[1][i] = -1;
				}
				 
			}			
		}
		else { 
			dp[1][i] = dp[1][i - 5] + 1;
		}
	}
	cout << dp[1][num] << endl;
}