题解 | 没有上司的舞会

#include <bits/stdc++.h>
using namespace std;
using ll = long long int;
using ld = long double;
using PII=pair<ll,ll>;
using PIII=pair<int,pair<int,int>>;
const ld ESP = 1e-10;
const ld PI = acosl(-1);
const int N=1e5+10;
const int M=2e5+10;
// const int mod = 1000000007;
const int mod = 998244353;
//随机化
random_device rd;
mt19937 gen(rd());
uniform_int_distribution<> dis(1, 1000000000);
// cout<<fixed<<setprecision(10);
int c[M];
int dp[M][2];
vector<vector<int>> e(M);
void dfs(int x,int f){
    dp[x][1]=c[x];
    dp[x][0]=0;
    for(auto y:e[x]){
        if(y==f)continue;
        else{
            dfs(y,x);
            dp[x][0]+=max({0,dp[y][1],dp[y][0]});
            dp[x][1]+=max(0,dp[y][0]);
        }
    }
}
void solve(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>c[i];
    }
    for(int i=1;i<n;i++){
        int u,v;
        cin>>u>>v; 
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dfs(1,0);
    // for(int i=1;i<=n;i++){
    //     cout<<dp[i][0]<<" "<<dp[i][1]<<'\n';
    // }
    cout<<max(dp[1][1],dp[1][0])<<'\n';
}

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int _=1;
    // cin>>_;
    while(_--){
        solve();
    }
    return 0;
}