洛谷—P1064 金明的预算方案 题解

题目链接： P1064 金明的预算方案
解题思路： 带有附件的背包问题，它属于01背包的变式。每一个物品最多只有两个附件，那么我们在对主件进行背包的时候，决策就不再是两个了，而是五个。
01背包的决策
1.不选，然后去考虑下一个
2.选，背包容量减掉那个重量，总值加上那个价值。
这个题的决策是五个，分别是：
1.不选，然后去考虑下一个
2.选且只选这个主件
3.选这个主件，并且选附件1
4.选这个主件，并且选附件2
5.选这个主件，并且选附件1和附件2.
代码：

#include<iostream>
using namespace std;
const int maxn = 500000;
int n, m;
int a[maxn], b[maxn], c[maxn];
int mw[maxn], mv[maxn];
int fw[maxn][3], fv[maxn][3],cnt[maxn],dp[maxn];
int main() {
	cin >> n >> m;
	for (int i = 1; i <= m; i++) {
		int x, y, z;
		cin >> x >> y >> z;
		if (!z) {
			mw[i] = x;
			mv[i] = x * y;
		}
		else {
			cnt[z]++;//记录每个主件的附件的数目
			fw[z][cnt[z]] = x;//记录当前的主件对应附件的价格
			fv[z][cnt[z]] = x * y;//记录当前的主件对应附件的价值
		}
	}
	for (int i = 1; i <= m; i++) {
		for (int j = n; j >= mw[i]; j--) {
			if (mw[i] == 0) {
				continue;
			}
			//01背包模板   
	        //每一个if的前提是背包能不能装下该物品   
		    //情况1：只要主件 和啥都不要比较   
			dp[j] = max(dp[j], dp[j - mw[i]] + mv[i]);
			//情况2：主件和附件1 和上面选出的较大值比较
			if (j >= (mw[i] + fw[i][1])) {
				dp[j] = max(dp[j], dp[j - (mw[i] + fw[i][1])] + mv[i] + fv[i][1]);
			}
			//情况3：主件和附件2 和上面选出的较大值比较   
			if (j >= (mw[i] + fw[i][2])) {
				dp[j] = max(dp[j], dp[j - (mw[i] + fw[i][2])] + mv[i] + fv[i][2]);
			}
			//情况4：都要 
			if (j >= (mw[i] + fw[i][1] + fw[i][2])) {
				dp[j] = max(dp[j], dp[j - (mw[i] + fw[i][1] + fw[i][2])] + mv[i] + fv[i][1] + fv[i][2]);
			}
		}
	}
	cout << dp[n] << endl;
}