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

> 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

---

> 解题思路
树状结构：ヽ(￣▽￣)و

利用链式储存结构。对于每一个顶点，开一条链，依次存储以该点为起点的边。下面的代码用e数组储存边的信息，head[i]储存i这个顶点对应的链的起始位置。通时e中的next域使所有起始点为i的边连成一条链。
（身为一个蒟蒻，当时我要是知道就不会暴搜了）

当这个点取的时候，他的所有儿子都不能取，所以
$f[i,1]=f[i][1]+f[e[i].y][0]的权值$
不取的时候，他的所有儿子取不取无所谓，不过当然应该取价值最大的一种情况。
所以
$f[now][0]+=max(f[e[i].y][1],f[e[i].y][0])$
这就是树状动规的基本形态。

看代码理解一下。。。 （我太难了）

---

> 代码

#include<iostream>
using namespace std;
int h[6005],n,r[6005],x1,y1,head[6005],s[6005]={0},f[6005][2],ans;//f[i][0]表示不选，f[i][1]表示选
struct{
	int x,y,next;//e.x为e.y的父节点
}e[6005];
void dp(int now)
{
      	f[now][1]=r[now];//如果选当前数的话就加上当前数的快乐指数
     	for(int i=head[now];i;i=e[i].next)
	{
	    dp(e[i].y);
		f[now][1]=f[now][1]+f[e[i].y][0];//当取这个点的时候，他的所有儿子都不能取
		f[now][0]+=max(f[e[i].y][1],f[e[i].y][0]);//当不取这个点的时候，取他的儿子中较大的
	}
}
//从根节点开始递归
int main(){
	cin>>n; 
	for(int i=1;i<=n;i++)
	cin>>r[i];
	for(int i=1;i<=n-1;i++)
	{
		cin>>x1>>y1;
		e[i].x=y1;
		e[i].y=x1;
		e[i].next=head[y1];
		head[y1]=i;
		s[x1]=1;
	}//利用链式储存结构,对于每一个顶点，开一条链，依次存储。
	cin>>x1>>y1;
	for(int i=1;i<=n;i++)
	if(s[i]==0){ans=i; break;}//找父节点
	dp(ans);
	cout<<max(f[ans][0],f[ans][1]);
	return 0;
}