Luogu P1352 没有上司的舞会

Description：

某大学有N个职员，编号为1~N。他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数Ri，但是呢，如果某个职员的上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。
接下来N-1行，每行输入一对整数L,K。表示K是L的直接上司。
最后一行输入0 0

Analysis：

樹形DP：f[i][0/1] := 以 i 爲根的子樹，不選（0）或選（1），的最大快樂指數。

\[$ 如果 i 不選，f[i][0] = \sigma_v max(f[v][0],f[v][1]) $ \]

\[$ 如果 i 選了，那兒子都不能選， f[i][1] = \sigma_v f[v][0] + r[i] $ \]

Code

#include<vector> 
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
vector<int> son[6010];
int f[6010][2],r[6010],c[6666],rt,n;
void dfs(int p)
{
	f[p][1] = r[p];
	f[p][0] = 0;
	for(int i = 0;i < son[p].size();++i)
	{
		int v = son[p][i];
		dfs(v);
		f[p][0] += max(f[v][1],f[v][0]);
		f[p][1] += f[v][0];
	}
}
int main()
{
	scanf("%d",&n);
	for(int i = 1;i <= n;++i)
	{
		scanf("%d",&r[i]);
	}
	for(int i = 1;i < n;++i)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		son[v].push_back(u);
		c[u]++;
	}
	for(int i = 1;i <= n;++i)
	{
		if(!c[i])
		{
			rt = i;
			break;
		}
	}
	int _0;
	scanf("%d%d",&_0,&_0);
	dfs(rt);
	printf("%d\n",max(f[rt][0],f[rt][1]));
	return 0;
}