【树形DP】有依赖的背包问题

有依赖的背包问题

题目链接：有依赖的背包问题

如果选择 物品5，则必须选择 物品1 和 物品2。

这是因为 物品2 是物品5 的 父节点，物品1 是 物品2 的 父节点。

求一个方案，使得该选择的物品合法，总体积不超过 V，且总价值最大。

我们考虑树形DP

对于结点i，对第 i 件物品只会依赖第i-1物品的状态
如果本题我们也采用该种状态依赖关系的话
对于结点i，我们需要枚举他所有子节点的组合 2^k 种可能
再枚举体积，最坏事件复杂度可能会达到 O(Nx2^NxV) 必然 TLE
换一种思考方式，枚举分给每个节点的 体积

#include <bits/stdc++.h>
using namespace std;
const int N=110;
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)
{
	for(int i=v[u];i<=m;i++)//只有体积容量大于当前结点的体积才能选择 
		f[u][i]=w[u];
	for(int i=h[u];~i;i=ne[i])
	{
		int son=e[i];
		dfs(son);
		for(int j=m;j>=v[u];j--)//一维优化 ，枚举体积 
			for(int k=0;k<=j-v[u];k++)//体积分割， 枚举该子节点在体积j下能使用的所有可能体积数
				f[u][j]=max(f[u][j],f[u][j-k]+f[son][k]);
	} 
}
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;
}