洛谷 1122 最大子树和

【题解】

　　直接做树形DP即可。f[i]表示取i及其子树的最大的美丽指数，若某个儿子son满足f[son]>0，f[i]就加上f[son].

#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
#define N 200010
#define rg register
using namespace std;
int n,m,ans=-2e9,tot,last[N],f[N];
struct edge{
    int to,pre;
}e[N];
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
} 
void dfs(int x,int fa){
    for(rg int i=last[x],to;i;i=e[i].pre)if((to=e[i].to)!=fa){
        dfs(to,x); if(f[to]>0) f[x]+=f[to];
    }
    ans=max(ans,f[x]);
}
int main(){
    n=read();
    for(rg int i=1;i<=n;i++) f[i]=read();
    for(rg int i=1;i<n;i++){
        int u=read(),v=read();
        e[++tot]=(edge){v,last[u]}; last[u]=tot;
        e[++tot]=(edge){u,last[v]}; last[v]=tot;
    }
    dfs(1,0);
    printf("%d\n",ans);
}