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

题目描述

某大学有 nn 个职员，编号为 1ldots n1…n。

他们之间有从属关系，也就是说他们的关系就像一棵以校长为根的树，父结点就是子结点的直接上司。

现在有个周年庆宴会，宴会每邀请来一个职员都会增加一定的快乐指数 r_iri​，但是呢，如果某个职员的直接上司来参加舞会了，那么这个职员就无论如何也不肯来参加舞会了。

所以，请你编程计算，邀请哪些职员可以使快乐指数最大，求最大的快乐指数。

输入格式

输入的第一行是一个整数 nn。

第 22 到第 (n + 1)(n+1) 行，每行一个整数，第 (i+1)(i+1) 行的整数表示 ii 号职员的快乐指数 r_iri​。

第 (n + 2)(n+2) 到第 2n2n 行，每行输入一对整数 l, kl,k，代表 kk 是 ll 的直接上司。

输出格式

输出一行一个整数代表最大的快乐指数。

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+100;
int n;
vector<int> g[maxn];
int dp[2][maxn];
//dp[0]表示不选择该节点的最优解
//dp[1]表示选择该节点的最优解
int a[maxn];
void dfs (int u,int pre) {
    dp[0][u]=0;
    dp[1][u]=a[u];
    for (int v:g[u]) {
        if (v==pre) continue;
        dfs(v,u);
        dp[0][u]+=max(dp[1][v],dp[0][v]);
        dp[1][u]+=dp[0][v];
    }
} 
int main () {
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",a+i);
    for (int i=1;i<n;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs(1,0);
    printf("%d
",max(dp[0][1],dp[1][1]));
}