生命之树(c++实现)

#include<iostream>
#include<cstring>
using namespace std;
const int N = 100010,M=N*2;
int n;
int w[N],h[N],ne[M],e[M],idx;
long long f[N];
void add(int a,int b){
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx;
    idx++;
}

void dfs(int u,int fa){
    f[u]=w[u];
    for(int i=h[u];i!=-1;i=ne[i]){
        int j = e[i];
        if(j!=fa) {
            dfs(j,u);
            f[u]+=max(0ll,f[j]);
        }
    }
}


int main(){
    scanf("%d",&n);
    memset(h,-1,sizeof h);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    
    for(int i=0;i<n-1;i++){
        int a,b;
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    dfs(1,-1);
    
    long long res = f[1];
    for(int i=2;i<=n;i++) res = max(res,f[i]);
    
    printf("%lld",res);
    
    
}