bzoj3829 [Poi2014]FarmCraft（树形dp+贪心）

f[x]表示x子树中最晚完成安装的时间的最小值。因为每条边只能走两遍，所以我们只能一个儿子一个儿子的遍历。
那么我们以何种顺序去遍历x的n个儿子呢？如果已经确定了一种顺序，那么
$f[x]=max\{\sum\limits_{k=1}^{i-1}sz[k]*2+f[i]+1|1<=i<=n\}$
对于i，j（i< j）两个位置的儿子，如果不交换比交换更优，则
$max(\sum\limits_{k=1}^{i-1}sz[k]*2+f[i]+1,\sum\limits_{k=1}^{j-1}sz[k]*2+f[j]+1)<max(\sum\limits_{k=1}^{i-1}sz[k]*2+f[j]+1,\sum\limits_{k=1}^{j-1}sz[k]*2+f[i]+1)$
即
$max(f[i],sz[i]*2+f[j])<max(f[j],sz[j]*2+f[i])$
以这个东西作为cmp即可。把儿子们排序后就是最优的了，计算答案即可。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 500010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,c[N],f[N],sz[N],a[N],h[N],num=0,fa[N];//f[x]-x子树中最晚完成安装的时间的最小值
struct edge{
    int to,next;
}data[N<<1];
inline bool cmp(int a,int b){return max(f[a],sz[a]*2+f[b])<max(f[b],sz[b]*2+f[a]);}
inline void dfs(int x){
    sz[x]=1;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;
        fa[y]=x;dfs(y);sz[x]+=sz[y];
    }int tot=0;f[x]=c[x];
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa[x]) continue;a[++tot]=y;
    }sort(a+1,a+tot+1,cmp);int tmp=0;
    for(int i=1;i<=tot;++i) f[x]=max(f[x],tmp*2+f[a[i]]+1),tmp+=sz[a[i]];
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();for(int i=1;i<=n;++i) c[i]=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1);
    printf("%d\n",max(f[1],(n-1)*2+c[1]));
    return 0;
}