7.9 树上背包3

最新推荐文章于 2024-10-16 19:08:56 发布

Henry_WYH

最新推荐文章于 2024-10-16 19:08:56 发布

阅读量148

点赞数

分类专栏：暑期训练文章标签：树形DP 有根树最大权值和动态规划 DFS

本文链接：https://blog.csdn.net/m0_51780913/article/details/125839785

版权

暑期训练专栏收录该内容

10 篇文章 0 订阅

订阅专栏

7.9

树上背包3

给你一个 n(1≤n≤1000)n(1≤n≤1000) 个点的有根树，其中1号点为根。每个点有一个权值 $a_i(1≤a_i≤10^5) 和重量 w_i(1≤w_i≤10^4)$

你需要选择一个重量恰好 kk 的包含根的连通块，最大的权值和。对所有的k=0,1,2,…, $m_k$ =0,1,2,…,m输出答案，其中 1≤m≤1041≤m≤104 。如果不存在，输出0。

从后往前做，令f[i][j]表示考虑DFS序中[i,n]这一段的节点，选的重量和不超过j的点集的最大权值和。

转移为f[i][j]=max(f[i][j], f[r[i]+1][j]+w[id[i]]),因为dfs序中r[id[i]]+1正好对应之后的那一个位置

const int N = 2010, M = 20100, INF = 0x3f3f3f3f;
int n,m,k,w[N],a[N],l[N],r[N],idx,id[N];
int f[N][M];
vector<int> edge[N];

void dfs(int u){
	l[++idx]=u,id[idx]=u;
	for(auto v:edge[u]){
		dfs(v);
	}
	r[u]=idx;
}

void solve(){
	n=read(),m=read();
	rep(i,2,n){
		int x=read();
		edge[x].push_back(i);
	}
	rep(i,1,n) a[i]=read();
	rep(i,1,n) w[i]=read();
	dfs(1); 
	for(int i=1;i<=m;++i) f[n+1][i]=-INF;
	for(int i=n;i>=1;--i){
		int u=id[i];
		for(int j=0;j<=m;++j){
			f[i][j]=f[r[u]+1][j];
			if(j>=w[u])
				f[i][j]=max(f[i][j], f[i+1][j-w[u]]+a[u]);
		}
	}
	int ans=0;
	for(int i=0;i<=m;++i)
		if(f[1][i]>=0)
			print(f[1][i]);
		else print(0);
}