AcWing 285. 没有上司的舞会

添加链接描述
树形dp

#include<bits/stdc++.h>
#define ll long long
#define ios ios::sync_with_stdio(false);cin.tie(0); cout.tie(0);
using namespace std;
const int M=6e3+2,N=3005;
int happy[M],dp[M][2];
bool st[M];
vector<int>g[M];
void dfs(int u)
{
    dp[u][1]=happy[u];
    for(int i=0;i<g[u].size();i++)
    {
        int v=g[u][i];
        dfs(v);
        dp[u][1]+=dp[v][0];
        dp[u][0]+=max(dp[v][0],dp[v][1]);
    }
}
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>happy[i];
    for(int i=0;i<n-1;i++)
    {
        int u,v;
        cin>>u>>v;
        g[v].push_back(u);
        st[u]=true;
    }
    int root=1;
    while(st[root]) root++;
    dfs(root);
    cout<<max(dp[root][0],dp[root][1])<<endl;
    return 0;
}