[luoguP1352] 没有上司的舞会（DP）

传送门

树上的dp，从底向上dp就行。

设dp[u][0]表示不选节点 u 的最大值，dp[u][1]表示选节点 u 的最大值。

则状态转移方程为：

dp[u][0] = ∑max(dp[v][1], dp[v][0])

dp[u][1] = ∑dp[v][0]  + val[u]

（节点v是节点u的孩子）

——代码

 #include <cstdio>
 #include <cstring>
 #include <iostream>
 #define MAXN 12001

 using namespace std;

 int n, cnt;
 int head[MAXN], to[MAXN], next[MAXN], val[MAXN], f[MAXN], r[MAXN], dp[MAXN][];

 inline void add(int x, int y)
 {
     to[cnt] = y;
     next[cnt] = head[x];
     head[x] = cnt++;
 }

 inline void dfs(int u)
 {
     int i, v;
     dp[u][] = val[u];
     for(i = head[u]; i != -; i = next[i])
     {
         v = to[i];
         if(v != f[u])
         {
             f[v] = u;
             dfs(v);
             dp[u][] += max(dp[v][], dp[v][]);
             dp[u][] += dp[v][];
         }
     }
 }

 int main()
 {
     int i, x, y, pos;
     scanf("%d", &n);
     memset(head, -, sizeof(head));
     for(i = ; i <= n; i++) scanf("%d", &val[i]);
     for(i = ; i < n; i++)
     {
         scanf("%d %d", &x, &y);
         add(y, x);
         r[x]++;
     }
     for(i = ; i <= n; i++)
         if(!r[i])
         {
             pos = i;
             break;
         }
     dfs(pos);
     printf("%d", max(dp[pos][], dp[pos][]));
     return ;
 }