https://www.acwing.com/problem/content/10/
// 10. 有依赖的背包问题.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//
#include <iostream>
#include <vector>
using namespace std;
/*
https://www.acwing.com/problem/content/10/
有 N 个物品和一个容量是 V 的背包。
物品之间具有依赖关系,且依赖关系组成一棵树的形状。如果选择一个物品,则必须选择它的父节点。
如下图所示:
QQ图片20181018170337.png
如果选择物品5,则必须选择物品1和2。这是因为2是5的父节点,1是2的父节点。
每件物品的编号是 i,体积是 vi,价值是 wi,依赖的父节点编号是 pi。物品的下标范围是 1…N。
求解将哪些物品装入背包,可使物品总体积不超过背包容量,且总价值最大。
输出最大价值。
输入格式
第一行有两个整数 N,V,用空格隔开,分别表示物品个数和背包容量。
接下来有 N 行数据,每行数据表示一个物品。
第 i 行有三个整数 vi,wi,pi,用空格隔开,分别表示物品的体积、价值和依赖的物品编号。
如果 pi=−1,表示根节点。 数据保证所有物品构成一棵树。
输出格式
输出一个整数,表示最大价值。
数据范围
1≤N,V≤100
1≤vi,wi≤100
父节点编号范围:
内部结点:1≤pi≤N;
根节点 pi=−1;
输入样例
5 7
2 3 -1
2 2 1
3 5 1
4 7 2
3 6 2
输出样例:
11
*/
const int N = 105;
int dp[N][N];
int n, v;
struct NODE {
int v, w, fa;
vector<int> son;
}nodes[N];
int root; int ans;
/*
20 50
18 23 10
4 31 10
1 27 4
22 21 7
16 1 10
22 31 14
7 39 -1
25 16 3
2 40 8
7 44 3
3 10 3
21 45 8
20 29 19
11 17 4
20 46 18
14 48 13
9 48 8
15 21 3
5 30 3
10 32 18
*/
void dfs(int x) {
for (int i = nodes[x].v; i <= v; i++) {
dp[x][i] = nodes[x].w;
}
int currv = nodes[x].v;
//以x为根节点能获取的最大获益
for (auto e : nodes[x].son) {
dfs(e);
for (int i = v; i >= currv; i--) {
int nextv = nodes[e].v;
for (int j = nextv; i + j <= v; j++) {
dp[x][i + j] = max(dp[x][i + j], dp[x][i] + dp[e][j]);
if (x == root) {
ans = max(ans, dp[x][i + j]);
}
}
}
}
return;
}
void solve() {
dfs(root);
cout << ans << endl;
}
int main()
{
cin >> n >> v;
for (int i = 1; i <= n; i++) {
cin >> nodes[i].v >> nodes[i].w >> nodes[i].fa;
if (-1 == nodes[i].fa) {
root = i;
}
else {
nodes[nodes[i].fa].son.push_back(i);
}
}
solve();
return 0;
}
标签:10,背包,int,依赖,物品,nodes,节点
From: https://www.cnblogs.com/itdef/p/18218165