代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;

const int N = 100005 ;

int n;
vector<int > e[N];
int w[N];
LL f[N];

void dfs(int u,int father){
    f[u]+=(LL)w[u];

    for(int i=0;i<e[u].size();i++){
        int j=e[u][i];
        if(j!=father){
            dfs(j,u);
            f[u]+=max(0ll,f[j]);
        }
    }
}

int main()
{

    cin >> n;
    for(int i=1;i<=n;i++) scanf("%d", &w[i]);
    for(int i=1;i<n;i++){
        int a,b;
        scanf("%d%d", &a, &b);
        e[a].push_back(b);
        e[b].push_back(a);
    }

    dfs(1,-1);

    LL res=f[1];
    for(int i=2;i<=n;i++) res=max(res,f[i]);

    cout << res;
    return 0;
}