bzoj1208 [HNOI2004]宠物收养所（平衡树）

就是平衡树求个前驱后继。记一下当前树上是人还是狗。

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 80010
#define mod 1000000
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,S=2,fa[N],c[N][2],sz[N],ans=0,owo=0,v[N],rt=0;
inline void update(int p){
    int l=c[p][0],r=c[p][1];
    sz[p]=sz[l]+sz[r]+1;
}
inline void rotate(int x,int &k){
    int y=fa[x],z=fa[y],l=x==c[y][1],r=l^1;
    if(y==k) k=x;
    else c[z][y==c[z][1]]=x;
    fa[c[x][r]]=y;fa[y]=x;fa[x]=z;
    c[y][l]=c[x][r];c[x][r]=y;update(y);update(x);
}
inline void splay(int x,int &k){
    while(x!=k){
        int y=fa[x],z=fa[y];
        if(y!=k){
            if(x==c[y][1]^y==c[z][1]) rotate(x,k);
            else rotate(y,k);
        }rotate(x,k);
    }
}
inline void ins(int &p,int x,int Fa){
    if(!p){p=++owo;v[p]=x;fa[p]=Fa;sz[p]=1;splay(p,rt);return;}
    if(x<=v[p]) ins(c[p][0],x,p);
    else ins(c[p][1],x,p);
}
inline void del(int p){
    splay(p,rt);int pre=c[rt][0],succ=c[rt][1];
    while(c[pre][1]) pre=c[pre][1];
    while(c[succ][0]) succ=c[succ][0];
    splay(pre,rt);splay(succ,c[pre][1]);c[succ][0]=fa[p]=0;
    update(succ);update(rt);
}
inline void solve(int x){
    ins(rt,x,0);splay(owo,rt);int pre=c[rt][0],succ=c[rt][1];
    while(c[pre][1]) pre=c[pre][1];
    while(c[succ][0]) succ=c[succ][0];
    if(pre==1) pre=succ;
    if(pre!=1&&succ!=2&&abs(v[succ]-x)<abs(v[pre]-x)) pre=succ;
    ans=(ans+abs(v[pre]-x))%mod;del(owo);del(pre);if(sz[rt]==2) S=2;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();ins(rt,-inf,0);ins(rt,inf,0);
    while(n--){
        int op=read(),x=read();
        if(S==2||S==op) ins(rt,x,0),S=op;
        else solve(x);
    }printf("%d\n",ans);
    return 0;
}