[树形DP] 猴腮雷 （没有上司的舞会）

题目描述 Description

新年就要来临啦，小猴腮雷一家N人将作为嘉宾被邀请到了春晚上。然而小猴腮雷的一家都曾经是大名鼎鼎的熊孩子，无论在什么时候都会，也只会和自己的直系亲属，也就是父母吵架，当然，即便是在春晚这个舞台上也是这样。为了到时候不会引起战争，组委会据此思考到底要邀请哪些人？更让组委会头疼是，每个猴腮雷拥有一个活泼值Ai，组委会既想要他们不会发生矛盾，也想尽量让来参加的猴腮雷活泼值的和最大?

输入 Input

第一行一个整数N。
接下来N行，第i+1行表示 i号猴腮雷的活泼值Ai。
接下来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

限制 Limits

对于70%的数据 1≤N≤1000
对于100%的数据 1≤N≤6000,−128≤Ai≤127
Time Limits : 1s & Memory Limit : 128MB

经典树形DP题，记dpx,0/1为猴腮雷x不来/来的情况，根据这个就能推出DP方程了：

dpx,0=∑son∈xmax{dpson,0,dpson,1}dpx,1=happyx+∑son∈xdpson,0

答案是max{dp1,0,dp1,1}。
时间复杂度O(n)。
Code