52 lines
1.8 KiB
C++
52 lines
1.8 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
|
|||
|
|
// 利用循环法进行分组
|
|||
|
|
const int N = 60, M = 32010;
|
|||
|
|
|
|||
|
|
struct Node {
|
|||
|
|
int w, v;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
int n, m;
|
|||
|
|
vector<Node> a[N]; // 主件
|
|||
|
|
vector<Node> b[N]; // 附件
|
|||
|
|
int f[M];
|
|||
|
|
|
|||
|
|
int main() {
|
|||
|
|
cin >> m >> n; // 容量上限，物品个数
|
|||
|
|
|
|||
|
|
for (int i = 1; i <= n; i++) {
|
|||
|
|
int w, p, q;
|
|||
|
|
cin >> w >> p >> q; // 体积,重要度，依赖哪个物品
|
|||
|
|
if (!q) // 如果是主件
|
|||
|
|
a[i].push_back({w, w * p}); // 记录主件i的列表中，有了only主件i这个东西
|
|||
|
|
else
|
|||
|
|
b[q].push_back({w, w * p}); // 记录主件q的列表中，有了一个附件
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
比如物品A是1个主件，2个附件，应该最后有4种组合： 主件；主件+附件1；主件+附件2; 主件+附件1+附件2;
|
|||
|
|
处理办法:
|
|||
|
|
① 第一步先放主件，把附件先存着；直到主件放完，再去放附件；
|
|||
|
|
② 循环的放；比如 主件+附件1+附件2=(主件+附件1)+附件2;也就是在前个物品基础之上再加上附件2
|
|||
|
|
*/
|
|||
|
|
for (int i = 1; i <= n; i++)
|
|||
|
|
for (auto c : b[i]) { // 遍历每个附件
|
|||
|
|
int w = c.w, v = c.v;
|
|||
|
|
int sz = a[i].size();
|
|||
|
|
for (int j = 0; j < sz; j++)
|
|||
|
|
a[i].push_back({a[i][j].w + w, a[i][j].v + v}); // 把扩展出来的组放在尾巴上
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 分组背包模板
|
|||
|
|
for (int i = 1; i <= n; i++)
|
|||
|
|
for (int j = m; j >= 0; j--)
|
|||
|
|
for (int k = 0; k < a[i].size(); k++) // 注意：这里k是从0开始，因为vector的特点决定
|
|||
|
|
if (j >= a[i][k].w)
|
|||
|
|
f[j] = max(f[j], f[j - a[i][k].w] + a[i][k].v);
|
|||
|
|
// 输出
|
|||
|
|
printf("%d\n", f[m]);
|
|||
|
|
return 0;
|
|||
|
|
}
|