【题解】CH5401 没有上司的舞会 树形DP

题目链接

描述

Ural大学有 $N$ 名职员，编号为 $1\sim N$。他们的关系就像一棵以校长为根的树，父节点就是子节点的直接上司。每个职员有一个快乐指数，用整数 $H_i$ 给出，其中 $1\le i\le N$。现在要召开一场周年庆宴会，不过，没有职员愿意和直接上司一起参会。在满足这个条件的前提下，主办方希望邀请一部分职员参会，使得所有参会职员的快乐指数总和最大，求这个最大值。

输入格式

第一行一个整数 $N$。$(1\le N\le 6000)$
接下来 $N$ 行，第 $i+1$ 行表示 $i$ 号职员的快乐指数 $H_i$。$(-128\le H_i\le 127)$
接下来 $N-1$ 行，每行输入一对整数 $L,K$。表示 $K$是 $L$ 的直接上司。
最后一行输入 $0,0$。

输出格式

输出最大的快乐指数。

样例输入

7
1
1
1
1
1
1
1
1 3
2 3
6 4
7 4
4 5
3 5
0 0

样例输出

5

---

设 $F[x,0]$ 表示从以 $x$ 为根的子树中邀请一部分职员参会，并且 $x$ 不参加舞会时，快乐指数总和的最大值。
$$F[x,0]=\sum _{s\in Son(x)}\max (F[s,0],F[s,1])$$
设 $F[x,1]$ 表示从以 $x$ 为根的子树中邀请一部分职员参会，并且 $x$ 参加舞会时，快乐指数总和的最大值。
$$F[x,1]=H[x]+\sum _{s\in Son(x)}F[s,0]$$

#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
const int N=1e4+10;
int n,f[N][2],vis[N],h[N];
vector<int>g[N];
void dp(int u)
{
	f[u][0]=0;
	f[u][1]=h[u];
	for(int i=0;i<g[u].size();i++)
	{
		int v=g[u][i];
		dp(v);
		f[u][0]+=max(f[v][0],f[v][1]);
		f[u][1]+=f[v][0];
	}
}
int main()
{
	//freopen("in.txt","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&h[i]);
    for(int i=1;i<n;i++)
    {
    	int l,k;
    	scanf("%d%d",&l,&k);
    	vis[l]=1;g[k].push_back(l);
	}
	int rt;
	for(int i=1;i<=n;i++)
	    if(!vis[i]){rt=i;break;}
	dp(rt);
	printf("%d\n",max(f[rt][0],f[rt][1]));
	return 0;
}

总结

经典的树形DP