P1352 没有上司的舞会（树形DP）

题目链接：没有上司的舞会 - 洛谷

这道题目可以说是树形DP的模板题了，由于上司与职员之间的关系图刚好是一棵树，所以我们令f[i][0]表示以i为根的子树中且i未参加的最大快乐指数，而f[i][1]表示以i为根的子树中且i参加的最大快乐指数，至于动态转移方程就是写在递归函数里面的，如果还有对常见树形DP更新方法不了解的小伙伴可以看下我之前的博客，我在里面详细介绍了树形DP的常见遍历方式，下面附上博客地址：常见树形dp递归更新方法_AC__dream的博客-CSDN博客，对于x的子节点j，可以进行如下更新：

f[x][0]代表x未参加，所以我们可以加上f[j][0]和f[j][1]中的较大值

f[x][1]代表x参加，所以我们只能加上f[j][0],

就按照这样的方式遍历完所有x的子节点即可完成对x的更新

初始化就是将f[x][1]初始化为a[x]，这个也比较好理解，下面上代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<queue>
using namespace std;
const int N=1e5+10;
typedef long long ll;
ll f[N][2];//f[i][0/1]表示以i为根的子树中i不参加/参加的最大快乐指数
ll a[N],ne[N],e[N],h[N],idx;
int vis[N];//为了找出最大的上司 
void add(int x,int y)
{
	e[idx]=y;
	ne[idx]=h[x];
	h[x]=idx++;
}
void dfs(int x,int fa)
{
	f[x][1]=a[x];
	for(int i=h[x];i!=-1;i=ne[i])
	{
		int j=e[i];
		if(j==fa) continue;
		dfs(j,x);
		f[x][0]+=max(f[j][0],f[j][1]);
		f[x][1]+=f[j][0];
	}
}
int main()
{
	int n;
	cin>>n;
	memset(h,-1,sizeof h);
	for(int i=1;i<=n;i++)
		scanf("%lld",&a[i]);
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		add(v,u);
		vis[u]=true;
	}
	int root;
	for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			root=i;
			break;
		}
	}
	dfs(root,-1);
	printf("%lld",max(f[root][0],f[root][1]));
	return 0;
}