洛谷 ~ P1352 ~ 没有上司的舞会 （树形DP）

思路

$dp[u][0]$表示不选 u 点以 u 根的树的快乐值，$dp[u][1]$表示选上 u 点以 u 根的树的快乐值。
$dp[u][0]=\sum max(dp[v][0],dp[v][1])$
$dp[u][1]=\sum dp[v][0]$

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 1e5+5;
const int INF = 0x3f3f3f3f;
int n, w[MAXN], dp[MAXN][2];
vector<int> G[MAXN];
void dfs(int u, int fa)
{
    dp[u][0] = 0;
    dp[u][1] = w[u];
    for (auto v: G[u])
    {
        if (v != fa)
        {
            dfs(v, u);
            dp[u][0] += max(dp[v][0], dp[v][1]);
            dp[u][1] += dp[v][0];
        }
    }
}
int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i++) scanf("%d", &w[i]);
    int u, v;
    while (~scanf("%d%d", &u, &v) && (u+v))
    {
        G[u].push_back(v);
        G[v].push_back(u);
    }
    dfs(1, -1);
    int ans = max(dp[1][0], dp[1][1]);
    printf("%d\n", ans);
    return 0;
}
/*
7
1 1 1 1 1 1 1
1 3
2 3
6 4
7 4
4 5
3 5
0 0
*/