bzoj2809: [Apio2012]dispatching

练习一下启发式合并，就照着别人写了一遍。（好菜QAQ）
从后往前枚举管理者，贪心选择最便宜的忍者，用splay+启发式合并维护。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>using namespace std;int read(){    char ch=getchar();int f=0;    while(ch<'0'||ch>'9') ch=getchar();    while(ch>='0'&&ch<='9') {f=f*10+(ch^48);ch=getchar();}    return f;}long long sum[1000005],l[1000005],mx,ans;int root[1000005],fa[1000005],ch[1000005][2],w[1000005],v[1000005],b[1000005],size[1000005];int n,cnt;void pushup(int x){    size[x]=size[ch[x][0]]+size[ch[x][1]]+1;    sum[x]=sum[ch[x][0]]+sum[ch[x][1]]+v[w[x]];}void rotate(int x,int &f){    int y=fa[x],z=fa[y],l,r;    l=(ch[y][1]==x);r=l^1;    if(z) ch[z][ch[z][1]==y]=x;    ch[y][l]=ch[x][r];ch[x][r]=y;    fa[ch[y][l]]=y;fa[y]=x;fa[x]=z;    pushup(y);pushup(x);    if(y==f) f=x;}void splay(int x,int &f){    while(x!=f)    {        int y=fa[x],z=fa[y];        if(y!=f)        {            if(ch[y][0]==x^ch[z][0]==y) rotate(x,f);else rotate(y,f);        }        rotate(x,f);    }}void insert(int &x,int f,int p,int k){    if(!x)    {        w[x=++cnt]=p;        fa[x]=f;        size[x]=1;        sum[x]=v[p];        splay(x,root[k]);        return ;    }    insert(ch[x][v[p]>v[w[x]]],x,p,k);}int find(int x,long long mx){    if(!x) return 0;    if(mx<sum[ch[x][0]]) return find(ch[x][0],mx);    int ans=size[ch[x][0]];    mx-=sum[ch[x][0]];    if(mx<v[w[x]]) return ans;    ans++,mx-=v[w[x]];    return ans+find(ch[x][1],mx);}void dfs(int x,int f){    if(!x) return;    insert(root[f],0,w[x],f);    dfs(ch[x][0],f);    dfs(ch[x][1],f);}void merge(int x,int y){    if(size[root[x]]<size[root[y]]) swap(root[x],root[y]);    dfs(root[y],x);}int main(){    n=read(),mx=read();    for(int i=1;i<=n;i++)    {        b[i]=read(),v[i]=read(),l[i]=read(),insert(root[i],0,i,i);    }    for(int i=n;i;i--)    {        int x=find(root[i],mx);        ans=max(ans,(long long)x*l[i]);        if(i!=1) merge(b[i],i);    }    cout<<ans;    return 0;}