【树形DP】没有上司的晚会

题目：

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

---

输入：

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

---

思路：

这题用树形DP…
首先，上司和职员是这样的关系：
上司去了，职员必定不去。上司不去，职员可去可不去，然后我们就设
DP方程

//（dep 是 一个参数）
//f[dep][0]为上司不去，下属选择去或不去
//f[dep][1]为上司去，下属必定不去
	f[dep][0] += max(f[e[i].p][0],f[e[i].p][1]);
	f[dep][1] += f[e[i].p][0];	

---

$Code$:

#include <cstdio>
#include <iostream>
using namespace std;
int n,l,k,root,ans;
int happy[100001],h[100001],f[100001][10];
bool jl[100001];
struct node
{
	int w,p,s;
}e[1000001];
void tree_DP(int dep)
{
	f[dep][1] = happy[dep];//初始化
	for(int i = h[dep]; i; i = e[i].s)
	{
	    tree_DP(e[i].p);
		f[dep][0] += max(f[e[i].p][0],f[e[i].p][1]);//DP
		f[dep][1] += f[e[i].p][0];		
	} 
}
int main ()
{
	scanf ("%d",&n);
	for (int i = 1; i <= n; ++i)
	    scanf("%d",&happy[i]);
	for (int i = 1; i <= (n - 1); ++i)
	{
		scanf ("%d%d",&l,&k);
		e[i].w = k,e[i].p = l,e[i].s =h[k],h[k] = i;//建树
		jl[l] = true;//记录一下
	}
	scanf("%d%d",&l,&k);
	for (int i = 1; i <= n; ++i)
	{
		if (jl[i] == false) //没被记录
		{
			root = i;//根
			break;//跳
		}
	}
	tree_DP(root);
	ans = max(f[root][0],f[root][1]);//答案
	printf("%d",ans);
	return 0;
}