Codevs 1794 修剪花卉

/*
    树形dp，刚开始总是wa三个点，后来使劲开大了数组才A
*/
#include<iostream>
#include<cstdio>
using namespace std;
#define maxn 16010<<4
int n,a[maxn],dp[maxn],head[maxn],num;
bool vis[maxn];
struct node{
    int to,pre;
}e[maxn];
void dfs(int u){
    if(dp[u])return;
    dp[u]=a[u];
    for(int i=head[u];i;i=e[i].pre){
        int to=e[i].to;
        if(!vis[to]){
            vis[to]=1;
            dfs(to);
            dp[u]+=max(0,dp[to]);
        }
    }
}
void Insert(int from,int to){
    e[++num].to=to;
    e[num].pre=head[from];
    head[from]=num;
}
int main(){
    int x,y;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        Insert(x,y);Insert(y,x);
    }
    vis[1]=1;
    dfs(1);
    int ans=0;
    for(int i=1;i<=n;i++)ans=max(ans,dp[i]);
    printf("%d",ans);
}