有依赖背包存在多棵树的情况
作者:
想不出来_6
,
2024-02-27 17:20:46
,
所有人可见
,
阅读 42
有依赖背包存在多棵树的情况,直接将0节点加入树中
#include <iostream>
#include <cstring>
using namespace std;
const int N = 111; // 将 N 的值增加1
int n, m, root;
int h[N], e[N], ne[N], idx;
int v[N], w[N];
int f[N][N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++ ;
}
void dfs(int u)
{
//先枚举所有状态体积小于等于j-v[u]的所有子节点们能够获得的最大价值
for (int i = h[u]; ~i; i = ne[i])
{
int son = e[i];
dfs(son); //从下往上算,先计算子节点的状态
for (int j = m - v[u]; j >= 0; -- j) //枚举所有要被更新的状态
{
for (int k = 0; k <= j; ++ k) //枚举该子节点在体积j下能使用的所有可能体积数
{
f[u][j] = max(f[u][j], f[u][j - k] + f[son][k]);
}
}
}
//最后选上第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; //清空没选上u的所有状态
}
int main()
{
memset(h, -1, sizeof h);
cin >> n >> m;
root = 0; // 将 root 的初始值设为 0
for (int i = 1; i <= n; ++ i)
{
int p;
cin >> v[i] >> w[i] >> p;
if (p == -1) add(root, i); // 如果 p 是 0,表示该节点没有父节点,也就是森林中的一棵树的根节点,那么就将该节点作为 0 的子节点
else add(p, i);
}
dfs(root);
cout << f[root][m] << endl;
return 0;
}