【SSL】1607&【洛谷】1352&【POJ】2342没有上司的晚会

【SSL】1607&【洛谷】1352&【POJ】2342没有上司的晚会

Time Limit:1000MS

Memory Limit:65536K

Description

Ural大学有N个职员，编号为1~N。他们有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。每个职员有一个快乐指数。现在有个周年庆宴会，要求与会职员的快乐指数最大。但是，没有职员愿和直接上司一起与会。

Input

第一行一个整数N。(1<=N<=6000)
接下来N行，第i+1行表示i号职员的快乐指数Ri。(-128<=Ri<=127)
接下来N-1行，每行输入一对整数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

思路

这题用树状动规。
当这个点取的时候，他的所有儿子都不能取，所以
f[i][1]=f[i所有儿子][0]的和＋i的权值。
不取的时候，他的所有儿子取不取无所谓，不过当然应该取价值最大的一种情况。
f[i][0]=所有max(f[i的儿子][1],f[i的儿子][0])的和。

代码

#include<iostream>
#include<cstdio> 
#include<cstring>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<iomanip>
using namespace std;
struct jgt
{
	int u,d,next;
};
jgt zg[6010];
int n,happy[6010],head[6010],f[6010][2],light[6010];
void input()
{
	int x,y,h=0,i;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%d",&happy[i]);
	scanf("%d%d",&x,&y);
	while(x!=0&&y!=0)//邻接表
	{
		h++;
		light[x]=1;
		zg[h].u=y;
		zg[h].d=x;
		zg[h].next=head[y];
		head[y]=h;
		scanf("%d%d",&x,&y);
	}
	return;
}
void DP(int w)
{
	int i;
	f[w][1]=happy[w];
	for(i=head[w];i;i=zg[i].next)
	{
		DP(zg[i].d);//求出孩子的值
		f[w][1]+=f[zg[i].d][0];//状态转移方程
		f[w][0]+=max(f[zg[i].d][0],f[zg[i].d][1]);
	}
	return;
}
int main()
{
	int i;
	input();
	for(i=1;i<=n;i++)//找根节点
	{
		if (!light[i])
		{
			DP(i);
			break;
		}
	}
	printf("%d",max(f[i][0],f[i][1]));
	return 0;
}