bzoj1208(宠物收养所 splay模板)

题目

#include<cstdio>
#include<iostream>
using namespace std;
typedef long long ll;
const int N=8e4+5,mod=1e6;
const ll INF=1e12;
int ch[N][2],f[N],sz[N],cnt[N],val[N],root=0,tot=0;
ll a[N];
inline void _clear(int x){
    ch[x][0]=ch[x][1]=f[x]=sz[x]=cnt[x]=val[x]=0;
}
inline void update(int x){
    if(!x) return;
    sz[x]=cnt[x]+sz[ch[x][0]]+sz[ch[x][0]];
}
inline void route(int x){
    int y=f[x],z=f[y],w=(ch[y][1]==x);
    ch[y][w]=ch[x][w^1],f[ch[y][w]]=y;
    ch[x][w^1]=y,f[y]=x;
    f[x]=z;
    if(z) ch[z][ch[z][1]==y]=x;
    update(y),update(x);
}
inline void splay(int x){
    for(int fa;fa=f[x];route(x))
        if(f[fa]) ((ch[fa][1]==x)^(ch[f[fa]][1]==fa)) ? route(x):route(fa);
    root=x;
}
inline void _insert(int x){
    if(!root){
        ++tot,root=tot;sz[tot]=cnt[tot]=1,val[tot]=x;
        ch[tot][0]=ch[tot][1]=f[tot]=0;
        return;
    }
    int now=root,fa=0;
    while(1)
    {
        if(x==val[now]){
            ++cnt[now];
            update(now),update(fa);splay(now);
            break;
        }
        fa=now,now=ch[now][x>val[now]];
        if(!now){
            ++tot;ch[tot][0]=ch[tot][1]=0,sz[tot]=cnt[tot]=1,val[tot]=x;
            ch[fa][x>val[fa]]=tot,f[tot]=fa;
            update(fa);splay(tot);
            break;
        }
    }
}
inline int _pre(){
    if(cnt[root]>1) return root;//假如有与root的值相同的值 前驱后继都是他自己
    int now=ch[root][0];
    while(ch[now][1]) now=ch[now][1];
    return now;
}
inline int _nex(){
    if(cnt[root]>1) return root;
    int now=ch[root][1];
    while(ch[now][0]) now=ch[now][0];
    return now;
}
inline void _delete(int x){
    if(cnt[root]>1) {--cnt[root];update(root);return;}
    if(!ch[root][0]&&!ch[root][1]) {_clear(root);root=0;return;}
    int tmp=root,ls=ch[root][0],rs=ch[root][1];
    if(!ls) {root=rs,f[root]=0,_clear(tmp);return;}
    if(!rs) {root=ls,f[root]=0,_clear(tmp);return;}
    splay(_pre());
    ch[root][1]=rs,f[rs]=root;
    _clear(tmp);update(root);
}

int main()
{
    int n,f,x;scanf("%d",&n);
    int numA=0,numB=0;ll sum=0;
    for(int i=1;i<=n;++i)//反正整个过程一定可以保证splay里面要么只有宠物 要么只有主人
    {
        scanf("%d%lld",&f,&a[i]);
        if(!f)
        {
            if(numB)//存在主人
            {
                _insert(a[i]);
                int dex1=_pre(),dex2=_nex();
                if(!dex1) a[dex1]=-INF;
                if(!dex2) a[dex2]=INF;
                _delete(a[i]);
                if(a[i]-a[dex1]<=a[dex2]-a[i]) splay(dex1),_delete(a[dex1]),sum+=a[i]-a[dex1];//md 注意 _delete之前一定确保要删除的值一定是root
                if(a[i]-a[dex1]>a[dex2]-a[i]) splay(dex2),_delete(a[dex2]),sum+=a[dex2]-a[i];
                a[0]=0;
                --numB,sum%=mod;
            }
            else ++numA,_insert(a[i]);
        }
        else//主人来了
        {
            if(numA)//存在宠物
            {
                _insert(a[i]);
                int dex1=_pre(),dex2=_nex();
                if(!dex1) a[dex1]=-INF;//注意没有前驱返回的是0 dex1 dex2不可能同时为0..
                if(!dex2) a[dex2]=INF;//同理没有后继会返回0  可能0和需要的更接近的 但0是无效的
                _delete(a[i]);
                if(a[i]-a[dex1]<=a[dex2]-a[i]) splay(dex1),_delete(a[dex1]),sum+=a[i]-a[dex1];
                if(a[i]-a[dex1]>a[dex2]-a[i]) splay(dex2),_delete(a[dex2]),sum+=a[dex2]-a[i];
                a[0]=0;//将a[0]还原 不还原也没事哦
                --numA,sum%=mod;
            }
            else ++numB,_insert(a[i]);
        }
    }
    printf("%lld\n",sum);
}