树形背包 学习笔记

树形背包

树形背包 顾名思义就是有根树上的背包问题；

树上的每个结点都有体积和价值，然后给你体积限制，问你最多可以获取的价值是多少；这里跟一般的线性背包问题没什么差别，但是树形背包一般还要求如果选择子节点物品，则必须选择父节点物品；

这个跟一般的树形dp是一样的，通过dfs回溯，用子节点更新父节点；

重点是这里的更新操作，一般是枚举父节点的容积，再枚举分配给子节点的容积；

上一道模板题：

洛谷 P2014 [CTSC1997]选课

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
#define ls k<<1
#define rs k<<1|1
#define inf 0x3f3f3f3f
using namespace std;
const int N=1000;
const int M=1000100;
const LL mod=100000000;
int n,m,head[N],cnt,w[N],in[N],dp[N][N];
struct Node{
	int to,nex;
}edge[N];
void add(int p,int q){
	edge[cnt].to=q;
	edge[cnt].nex=head[p];
	head[p]=cnt++;
}
void dfs(int p){
	dp[p][1]=w[p];
	for(int i=head[p];~i;i=edge[i].nex){
		int q=edge[i].to;
		dfs(q);
		for(int j=m;j>=1;j--){
			for(int k=j-1;k>=1;k--){
				dp[p][j]=max(dp[p][j],dp[p][j-k]+dp[q][k]);
			}
		}
	}
}
int main(){
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){
		int k,s;
		scanf("%d%d",&k,&s);
		if(k) add(k,i),in[i]++;
		w[i]=s;
	}
	for(int i=1;i<=n;i++){
		if(in[i]==0) add(0,i);
	}
	m++;
	dfs(0);
	cout<<dp[0][m]<<endl;
    return 0;
}