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

题目传送门

思路

购买附件是一定要购买主件，并且附件最多有两个，所以一个主件搭配附件的方法只有四种（选第一个，选第二个，都选，不选）。所以我们在dp时只考虑主件，枚举这四种方案。

如何存储主件的附件？

输入中的第三列表示的不是第几个主件，而是在第几行的主件。我们可以用一个二维数组，第一维表示几行的主件，第二维表示这个主件的附件在第几行。

代码

#include<bits/stdc++.h>
#define endl '\n';
using namespace std;
int n,m,k;
const int N=100,M=1e4*3.2;
int w[N],v[N],f[N],dp[M],c[M];
int e[N][5];
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=m;i++){
		cin>>w[i]>>v[i]>>f[i];
		if(f[i]){
			if(e[f[i]][0]){
				e[f[i]][1]=i;
			}else e[f[i]][0]=i;
		}
	}
	for(int i=1;i<=m;i++){
		for(int j=n;j>=w[i];j-=10){
			if(f[i]==0){
				dp[j]=max(dp[j],dp[j-w[i]]+v[i]*w[i]);
				if(j>=w[i]+w[e[i][0]])
				dp[j]=max(dp[j],dp[j-w[i]-w[e[i][0]]]+v[i]*w[i]+w[e[i][0]]*v[e[i][0]]);
				if(j>=w[i]+w[e[i][1]])
				dp[j]=max(dp[j],dp[j-w[i]-w[e[i][1]]]+v[i]*w[i]+w[e[i][1]]*v[e[i][1]]);
				if(j>=w[i]+w[e[i][0]]+w[e[i][1]])
				dp[j]=max(dp[j],dp[j-w[i]-w[e[i][1]]-w[e[i][0]]]+w[e[i][1]]*v[e[i][1]]+w[e[i][0]]*v[e[i][0]]+w[i]*v[i]);
			}	
		}
	}
	cout<<dp[n];
	return 0;
}