没有上司的晚会

Description

有个公司要举行一场晚会。 
为了能玩得开心，公司领导决定：如果邀请了某个人，那么一定不会邀请他的上司 
（上司的上司，上司的上司的上司……都可以邀请）。  
每个参加晚会的人都能为晚会增添一些气氛，求一个邀请方案，使气氛值的和最大。

Input

第1行一个整数N（1<=N<=6000）表示公司的人数。 
接下来N行每行一个整数。第i行的数表示第i个人的气氛值x(-128<=x<=127)。 
接下来每行两个整数L，K。表示第K个人是第L个人的上司。 
输入以0 0结束。

Output

一个数，最大的气氛值和。

Sample Input

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

Sample Output

5

Source

ural 1039

【分析】

        经典的树型DP。

        我们这样定义状态：f[i][0]表示以i为根的子树的最大气氛值，且不选i这个节点

                                            f[i][1]表示以i为根的子树的最大气氛值，且选i这个节点

        很容易的到DP方程：

                                            f[i][0]=sigma（max(0,f[son][0],f[son][1])）；f[i][1]=sigma(max(0,f[son][0]))+max(0,v[i])；

        注意边界条件：

                                            f[i][1]=max(0,v[i]); i为叶子节点

        那么答案就是：

                                            max(f[root][0],f[root][1]);

【代码】

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int N,v[6005];
int f[6005][2],du[6005],root;
int y[6006],next[6006],last[6006],tot;
void _init()
{
	scanf("%d",&N);
	for(int i=1;i<=N;i++)
	    scanf("%d",&v[i]);
	int xt,yt;
	scanf("%d%d",&xt,&yt);
	while(xt&&yt)
	{
		du[xt]++;
		tot++;
		y[tot]=xt;
		next[tot]=last[yt];
		last[yt]=tot;
		scanf("%d%d",&xt,&yt);
	}
}
void _DFS(int x)
{
	int t0=0,t1=0;
	for(int j=last[x];j;j=next[j])
	{
		_DFS(y[j]);
		t0+=max(0,max(f[y[j]][0],f[y[j]][1]));
		t1+=max(0,f[y[j]][0]);
	}
	f[x][0]=max(f[x][0],t0);
	f[x][1]=max(f[x][1],t1+max(0,v[x]));
}
void _solve()
{
	for(int i=1;i<=N;i++)  //找到根节点
		if(!du[i])
		{
			root=i;
			break;
		}
	for(int i=1;i<=N;i++)  //边界条件
		if(du[i]==1)
		    f[i][1]=max(0,v[i]);
	_DFS(root);
	printf("%d\n",max(f[root][0],f[root][1]));
}
int main()
{
	_init();
	_solve();
	return 0;
}