【ybtoj高效进阶5-4-1】树上求和

最新推荐文章于 2022-08-24 15:23:10 发布

SSL_DFT

最新推荐文章于 2022-08-24 15:23:10 发布

阅读量122

点赞数

分类专栏：树形数据结构动态规划文章标签：树形DP

本文链接：https://blog.csdn.net/m0_51796369/article/details/119190226

版权

树形数据结构同时被 2 个专栏收录

19 篇文章 0 订阅

订阅专栏

动态规划

17 篇文章 0 订阅

订阅专栏

这篇博客介绍了一种使用树形动态规划（树形DP）解决树上节点求和最大化的算法。给定一棵树，目标是选择一些节点，使得它们的和最大，但条件是选择了父节点就不能选择其子节点。通过定义两个状态表示是否选择当前节点，采用逆推方式从叶子节点处理到根节点，最后输出根节点的最大和作为答案。

摘要由CSDN通过智能技术生成

【ybtoj高效进阶5-4-1】树上求和

题目大意：

给你一棵树，给出节点的父子关系，让你从中选出一些节点，让他们的和最大，然后如果选择了父节点就不能选择他们的子节点，问这个最大的和是多少？

思路：

树形DP，我们要用逆推，，从叶子节点一直处理到根节点。
我们设 $f [i] [0]$ 表示处理到第 i 个点时不选这个点的最大值， $f [i] [1]$ 表示选了这个点的最大值，son(i)是i的子节点集合。
那么取了该节点我们就不能取它的子节点，如果没取那么就可取可不取，
所以状态转移方程是： $f [i] [0] + = m a x (f [j] [0], f [j] [1])$ $f [i] [1] + = f [j] [0]$ $其中j\in son(i)$
那么最后的的答案就是 $m a x (f [r o o t] [0] ， f [r o o t] [1])$ ，其中 $r o o t$ 是这棵树的根节点

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#define r register
#define rep(i,x,y) for(r ll i=x;i<=y;++i)
#define per(i,x,y) for(r ll i=x;i>=y;--i)
using namespace std;
typedef long long ll;
const ll V=6*1e3+10;
struct node
{
	ll val,son[V],tot;
}tree[V];
ll n,x,y,f[V][2],fa[V];
void dp(ll now)
{
	f[now][1]=tree[now].val;
	f[now][0]=0;
	if(tree[now].tot==0) return ;
	rep(i,1,tree[now].tot) //tot[i]:i节点子节点的个数
	{
		ll y=tree[now].son[i];
		dp(y); //逆推从后往前，从叶子节点向根节点
		f[now][0]+=max(f[y][0],f[y][1]);
		f[now][1]+=f[y][0];
	}
}
int main()	
{
	scanf("%lld",&n);
	rep(i,1,n)
	 scanf("%lld",&tree[i].val);
	rep(i,1,n-1)
	{
		scanf("%lld%lld",&x,&y);
		tree[y].son[++tree[y].tot]=x; //记录子节点
		++fa[x];
	}
	rep(i,1,n)
	 if(fa[i]==0) //根节点
	 {
	 	dp(i);
	 	printf("%lld",max(f[i][0],f[i][1]));
	 	break ;
	 } 
	return 0;
}