原题传送门
思路:因为每个主件最多只有两个附件,所以枚举每个主件如果满足条件的话可以分为五种情况购买:什么都不买;只买主件;买主件和第一个附件;买主件和第二个附件;买主件和两个附件。然后分别计算出每种情况的物品价格与重要度乘积之和,最后取最大值即可.
代码参考
#include <bits/stdc++.h>
using namespace std;
int v[32001][3], p[61][3],f[32001];
/*
** v[i][j]代表第i个物品的第j个附件的价格
** p[i][j]代表第i个物品的第j个附件的重要度
** f[i]代表花了i元时,所买物品的价格与重要度乘积之和的最大值
*/
int cost2(int i,int x, int y){
return v[i][x] + v[i][y];
}
int cost3(int i,int x, int y, int z){
return v[i][x] + v[i][y] + v[i][z];
}
int vp(int i,int x){
return v[i][x] * p[i][x];
}
int main() {
int n,m;
cin>>n>>m;
for (int i = 1; i <= m; i++) {
int _v, _p, _q;//每件物品的 _v:价格,_p:重要度,_q:对应的主件
cin>>_v>>_p>>_q;
if (!_q){
// 是主件
v[i][0] = _v;
p[i][0] = _p;
}else if(v[_q][1] == 0){
// 是第一个附件
v[_q][1] = _v;
p[_q][1] = _p;
}else{
// 是第二个附件
v[_q][2] = _v;
p[_q][2] = _p;
}
}
for(int i = 1; i <= m; i++)
for (int j = n; j >= 0; j--) {
if (j >= v[i][0]) // 够买主件
f[j] = max(f[j], f[j - v[i][0]] + vp(i,0));
if (j >= cost2(i,0, 1)) // 还够买第一个附件
f[j] = max(f[j], f[j - cost2(i,0, 1)] + vp(i,0) + vp(i,1));
if (j >= cost2(i,0, 2)) // 还够买第二个附件
f[j] = max(f[j], f[j - cost2(i,0, 2)] + vp(i,0) + vp(i,2));
if (j >= cost3(i,0, 1, 2)) // 还够买两个附件
f[j] = max(f[j], f[j - cost3(i,0, 1, 2)] + vp(i,0) + vp(i,1) + vp(i,2));
}
cout<<f[n];
}