题目传送门
思路
购买附件是一定要购买主件,并且附件最多有两个,所以一个主件搭配附件的方法只有四种(选第一个,选第二个,都选,不选)。所以我们在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;
}