BZOJ 2809 [Apio2012]dispatching

【题解】

　　贪心地选择子树内的节点，优先选代价小的。如果当前费用超出限制了就依次去掉当前费用最大的点，直到费用不超过限制。每个点的情况由它的孩子合并得到，所以要用到可并堆（左偏树）。

　　

#include<cstdio>
#include<algorithm>
#define LL long long
#define rg register
#define N 200010
using namespace std;
int tot,root,last[N],c[N][2],rt[N],size[N];
LL n,m,sum[N],val[N],l[N],ans=0;
struct edge{
    int to,pre;
}e[N];
inline LL read(){
    LL 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;
}
inline LL max(LL x,LL y){return x>y?x:y;}
inline void update(int x){
    size[x]=size[c[x][0]]+size[c[x][1]]+1;
    sum[x]=sum[c[x][0]]+sum[c[x][1]]+val[x];
}
inline int merge(int x,int y){
    if(!x||!y) return x+y; if(val[x]<val[y]) swap(x,y);
    c[x][1]=merge(c[x][1],y); swap(c[x][0],c[x][1]);
    update(x); return x;
}
inline void pop(int &x){
    int tmp=merge(c[x][0],c[x][1]);
    c[x][0]=c[x][1]=0; size[x]=1; sum[x]=val[x]; x=tmp;
}
void dfs(int x){
    rt[x]=x;
    for(rg int i=last[x],to;i;i=e[i].pre) dfs(to=e[i].to),rt[x]=merge(rt[x],rt[to]);
    while(size[rt[x]]&&sum[rt[x]]>m) pop(rt[x]);
    ans=max(ans,1ll*size[rt[x]]*l[x]);
}
int main(){
    n=read(); m=read();
    for(rg int i=1;i<=n;i++){
        LL x=read(),y=read(),z=read();
        val[i]=sum[i]=y; l[i]=z; size[i]=1;
        if(x) e[++tot]=(edge){i,last[x]},last[x]=tot; else root=i;
    }
    dfs(root);
    printf("%lld\n",ans);
    return 0;
}