题目链接:没有上司的舞会 - 洛谷
这道题目可以说是树形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;
}