[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][2];

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][1] = val[u];
    for(i = head[u]; i != -1; i = next[i])
    {
        v = to[i];
        if(v != f[u])
        {
            f[v] = u;
            dfs(v);
            dp[u][0] += max(dp[v][1], dp[v][0]);
            dp[u][1] += dp[v][0];
        }
    }
}

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