本文介绍一个简短好写的数据结构,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;
}