题目传送门
算法
(\(DP\),分组背包问题)
可以将每个主件及其附件看作一个物品组,记主件为 \(p\),两个附件为 \(a\),\(b\),则最多一共有\(4\)种组合:
\(p\)
\(p\),\(a\)
\(p\),\(b\)
\(p\),\(a\),\(b\)
这四种组合是互斥的,最多只能从中选一种,因此可以将每种组合看作一个物品,那么问题就变成了分组背包问题。可以参考\(AcWing\) \(9\). 分组背包问题。
在枚举四种组合时可以使用二进制的思想,可以简化代码。
#include <bits/stdc++.h>
using namespace std;
const int N = 70; //希望购买物品的个数上限
const int M = 32010; //总钱数上限
//描述每个物品的结构体
struct Node {
int v;
int w;
};
int n; //希望购买物品的个数
int m; //总钱数
Node a[N]; //主物品数组
vector<Node> g[N]; //二维的数组,第一维是主件的号码,第二维是它的附件有哪些
int f[M]; //一维的分组背包DP数组
int main() {
//优化输入
ios::sync_with_stdio(false);
cin >> m >> n;
for (int i = 1; i <= n; i++) {
/*
v表示i物品的价格,p表示i物品的重要度(1~5)。
q表示i物品是主件还是附件。如果q=0,表示该物品为主件,
如果q>0,表示该物品为附件,q是所属主件的编号:parent_id概念
*/
int v, p, q;
cin >> v >> p >> q;
p *= v;//使每件物品的价格与重要度的乘积的总和最大
//i是主件
if (!q) a[i] = {v, p};
//i是附件,需记录到主件与附件的关系数组中去
else g[q].push_back({v, p});
}
//分组背包+二进制组合表示
for (int i = 1; i <= n; i++)
for (int u = m; u >= 0; u--)
//遍历每一个可能的组合方式
//(1)p (2) p a (3)p b (4) p a b
//a b-> 0 0,0 1,1 0,1 1 共四种
for (int j = 0; j < 1 << g[i].size(); j++) {
//当前物品i的体积和价值
int v = a[i].v, w = a[i].w;
for (int k = 0; k < g[i].size(); k++)
if (j >> k & 1) {//当前二进制数此位置为1
v += g[i][k].v;
w += g[i][k].w;
}
//如果可能装的下,那么替换更新
if (u >= v) f[u] = max(f[u], f[u - v] + w);
}
//输出结果
printf("%d", f[m]);
return 0;
}