POJ 2342 Anniversary party (树dp)

题目链接：http://poj.org/problem?id=2342

有n个人，每个人有活跃值。下面n-1行u和v表示u的上司是v，有直接上司和下属的关系不能同时参加party，问你party最大的活跃值是多少。

也就是说一棵树中，选择的点不能是相邻的点，且要使活跃值最大。

简单的树形dp，任意选一个点开始遍历，从叶子节点开始回溯。

dp[i][0]表示不选i节点最大的活跃度，则dp[i][1]表示选i节点最大的活跃度。

i与j相连，dp[i][0] += max(dp[j][0], dp[j][1]);   dp[i][1] += dp[j][0];

 //#pragma comment(linker, "/STACK:102400000, 102400000")
 #include <algorithm>
 #include <iostream>
 #include <cstdlib>
 #include <cstring>
 #include <cstdio>
 #include <vector>
 #include <cmath>
 #include <ctime>
 #include <list>
 #include <set>
 #include <map>
 using namespace std;
 typedef long long LL;
 typedef pair <int, int> P;
 const int N = 1e4 + ;
 struct Edge {
     int next, to;
 }edge[N << ];
 int dp[N][];
 int a[N], cnt, head[N];

 inline void add(int u, int v) {
     edge[cnt].to = v;
     edge[cnt].next = head[u];
     head[u] = cnt++;
 }

 void dfs(int u, int p) {
     dp[u][] = ;
     dp[u][] = a[u];
     for(int i = head[u]; ~i; i = edge[i].next) {
         int v = edge[i].to;
         if(v == p)
             continue;
         dfs(v, u);
         dp[u][] += max(dp[v][], dp[v][]);
         dp[u][] += dp[v][];
     }
 }

 int main()
 {
     int n, u, v;
     while(~scanf("%d", &n)) {
         for(int i = ; i <= n; ++i)
             scanf("%d", a + i);
         memset(head, -, sizeof(head));
         cnt = ;
         for(int i = ; i < n; ++i) {
             scanf("%d %d", &u, &v);
             add(u, v);
             add(v, u);
         }
         scanf("%d %d", &u, &v); //0 0
         dfs(, -);
         printf("%d\n", max(dp[][], dp[][]));
     }
     return ;
 }