AcWing 10. 有依赖的背包问题--代码注释
原题链接
困难
作者:
zqw
,
2024-10-08 21:13:40
,
所有人可见
,
阅读 2
#include <bits/stdc++.h>
using namespace std;
const int N = 110;
int n, m;
int h[N], e[N], ne[N], w[N], v[N], idx;
int f[N][N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs(int u)
{
// part1:处理以u为根的子树中,子树部分
for (int i = h[u]; ~i; i = ne[i]) //对每个子树做分组背包问题,划分的依据是分配的可用体积
{
int t = e[i];
dfs(t);
for (int j = m - v[u]; j >= 0; j -- ) //遍历,以u为根的子树在分配不同体积时可达到的最大价值
for (int k = 0; k <= j; k ++ ) //其中分给以t为根的这个子树k的体积
f[u][j] = max(f[u][j], f[u][j - k] + f[t][k]);
}
//part2:处理u这个根节点
for (int j = m; j >= v[u]; j -- ) f[u][j] = f[u][j - v[u]] + w[u];
for (int j = 0; j < v[u]; j ++ ) f[u][j] = 0;
}
int main()
{
cin >> n >> m;
memset(h, -1, sizeof h);
int root;
for (int i = 1; i <= n; i ++ )
{
int p;
cin >> v[i] >> w[i] >> p;
if (p == -1) root = i;
else add(p, i);
}
dfs(root);
cout << f[root][m] << endl;
return 0;
}