普通平衡树（bzoj 3224）

Description

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

Input

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)

Output

对于操作3,4,5,6每行输出一个数，表示对应答案

Sample Input

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

Sample Output

106465
84185
492737

HINT

1.n的数据范围：n<=100000

2.每个数的数据范围：[-1e7,1e7]

数据如下http://pan.baidu.com/s/1jHMJwO2

/*

  一个平衡树的模板写了一上午了，能犯的错误基本都犯了。。。

  首先没加哨兵，再就是删除操作时没有更新sz（这个巨坑），真是长记性了。

*/

#include<cstdio>

#include<iostream>

#define N 100010

using namespace std;

int son[N][],fa[N],val[N],cnt[N],sz[N],m,rt,size;

void pushup(int x){

    sz[x]=sz[son[x][]]+sz[son[x][]]+cnt[x];

}

void rotate(int x,int &k){

    int y=fa[x],z=fa[y],l,r;

    if(son[y][]==x) l=;else l=;r=l^;

    if(y==k) k=x;

    else {

        if(son[z][]==y) son[z][]=x;

        else son[z][]=x;

    }

    fa[x]=z;fa[y]=x;fa[son[x][r]]=y;

    son[y][l]=son[x][r];son[x][r]=y;

    pushup(y);pushup(x);

}

void splay(int x,int &k){

    while(x!=k){

        int y=fa[x],z=fa[y];

        if(y!=k){

            if((son[y][]==x)^(son[z][]==y)) rotate(x,k);

            else rotate(y,k);

        }

        rotate(x,k);

    }

}

void insert(int v){

    int k=rt,y=;

    while(k&&val[k]!=v) y=k,k=son[k][v>val[k]];

    if(k) cnt[k]++;

    else {

        k=++size;sz[k]=;cnt[k]=;fa[k]=y;val[k]=v;

        if(y) son[y][v>val[y]]=k;

    }

    splay(k,rt);

}

void find1(int v){//查找v的位置

    int k=rt;if(!k)return;

    while(son[k][v>val[k]]&&val[k]!=v)

      k=son[k][v>val[k]];

    splay(k,rt);

}

int find2(int x){//查找排名为x的数

    x++;

    int k=rt;

    if(sz[k]<x)return ;

    while(){

        if(sz[son[k][]]<x&&sz[son[k][]]+cnt[k]>=x) return k;

        if(sz[son[k][]]>=x) k=son[k][];

        else x-=(sz[son[k][]]+cnt[k]),k=son[k][];

    }

    return k;

}

int nxt(int x,int f){

    find1(x);

    if((val[rt]>x&&f)||val[rt]<x&&!f) return rt;

    int p=son[rt][f];

    while(son[p][f^]) p=son[p][f^];

    return p;

}

void del(int v){

    find1(v);

    int x=rt,k;

    if(cnt[x]>){cnt[x]--;sz[x]--;return;}

    if(!son[x][]||!son[x][]){

        rt=son[x][]+son[x][];

    }

    else {

        k=son[x][];

        while(son[k][])k=son[k][];sz[k]+=sz[son[x][]];

        fa[son[x][]]=k;son[k][]=son[x][];

        rt=son[x][];

    }

    fa[rt]=;splay(k,rt);

}

int main(){

    insert(-0x7fffffff);insert(0x7fffffff);

    scanf("%d",&m);

     for(int i=;i<=m;i++){

        int opt,x;scanf("%d%d",&opt,&x);

        if(opt==) insert(x);

        if(opt==) del(x);

        if(opt==) find1(x),printf("%d\n",sz[son[rt][]]);

        if(opt==) printf("%d\n",val[find2(x)]);

        if(opt==) printf("%d\n",val[nxt(x,)]);

        if(opt==) printf("%d\n",val[nxt(x,)]);

    }

    return ;

}

巴特西