【Luogu】P3574FAR_FarmCraft（树形贪心）

　　题解链接

　　想了一个错的贪心爆零了，气死。

　　题目链接

#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#include<cstring>
#define maxn 500050
using namespace std;

inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}

struct Edge{
    int next,to;
}edge[maxn*];
int head[maxn],num;
inline void add(int from,int to){
    edge[++num]=(Edge){head[from],to};
    head[from]=num;
}

long long q[maxn];
long long f[maxn];
long long s[maxn];
int sta[maxn],top;

bool cmp(int a,int b){    return f[a]>f[b];    }

void dfs(int x,int fa){
    int ret=top;
    for(int i=head[x];i;i=edge[i].next){
        int to=edge[i].to;
        if(to==fa)    continue;
        dfs(to,x);
        s[x]+=+s[to];
        sta[++top]=to;
    }
    if(x^)    f[x]=q[x]-s[x];
    int now=s[x];
    sort(sta+ret+,sta+top+,cmp);
    for(int i=ret+;i<=top;++i){
        int to=sta[i];
        now-=s[to]+;
        f[x]=max(f[x],f[to]-now-);
    }
    f[x]=max(f[x],0LL);
    top=ret;
}

int main(){
    int n=read();
    for(int i=;i<=n;++i)    q[i]=read();
    for(int i=;i<n;++i){
        int from=read(),to=read();
        add(from,to);
        add(to,from);
    }
    dfs(,);
    printf("%lld\n",max(f[],q[])+n*-);
    return ;
}