BZOJ 3224 普通平衡树 Trie树

本文介绍一个简短好写的数据结构，01trie树，它可以支持普通平衡树所要求所有操作，而且简单易懂。

代码原作者imone

对于每个数进行二进制拆分，把拆成的01串插进Trie树中。
由于负数的二进制处理比较麻烦，所以我们不如把其补成正整数，询问时再减去即可。
其余的见代码吧，相当好理解。

#include <cstdio>
#define ls(x) (ch[x][0])
#define rs(x) (ch[x][1])
const int maxn=120000*24;
int ch[maxn][2],size[maxn],cnt,root=++cnt,x,op;
int n,fix=1e7,full=24;
void Insert(int x,int add){x+=fix;
    int rt=root;
    for(int i=full;~i;i--){
        bool op=x>>i&1;
        if(!ch[rt][op]) ch[rt][op]=++cnt;
        rt=ch[rt][op];
        size[rt]+=add;
    }
}
int Rank(int x){x+=fix;
    int res=0,rt=root;
    for(int i=full;~i;i--){
        bool op=x>>i&1;
        if(op)res+=size[ls(rt)],rt=rs(rt);
        else rt=ls(rt);
    }
    return res;
}
int kth(int k){
    int res=0,rt=root;
    for(int i=full;~i;i--){
        if(k>size[ls(rt)])k-=size[ls(rt)],res|=1<<i,rt=rs(rt);
        else rt=ls(rt);
    }
    return res-fix;
}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&op,&x);
        if(op==1)Insert(x,1);
        else if(op==2)Insert(x,-1);
        else if(op==3)printf("%d\n",Rank(x)+1);
        else if(op==4)printf("%d\n",kth(x));
        else if(op==5)printf("%d\n",kth(Rank(x)));
        else if(op==6)printf("%d\n",kth(Rank(x+1)+1));
    }
    return 0;
}