posted on 2022-01-17 16:13:39 | under 模板 | source
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
template<int N,int M=2,int MAXD=30> struct trie{
int ch[N+10][M],tot,rt,siz[N+10];
trie():tot(0),rt(newnode()){siz[0]=0,insert(-1e7),insert(1e7);}
int newnode(){return memset(ch[++tot],0,sizeof ch[tot]),siz[tot]=0,tot;}
void _insert(int x,int& p,int v){
if(!p) p=newnode();
if(v==-1) return siz[p]++,void();
_insert(x,ch[p][x>>v&1],v-1);
siz[p]=siz[ch[p][0]]+siz[ch[p][1]];
}
void _remove(int x,int& p,int v){
if(!p) p=newnode();
if(v==-1) return siz[p]&&siz[p]--,void();
_remove(x,ch[p][x>>v&1],v-1);
siz[p]=siz[ch[p][0]]+siz[ch[p][1]];
}
int _getrank(int x,int& p,int v){
if(!p) p=newnode();
if(v==-1) return 0;
if(x>>v&1) return siz[ch[p][0]]+_getrank(x,ch[p][1],v-1);
else return _getrank(x,ch[p][0],v-1);
}
int _getkth(int x,int& p,int v){
if(!p) p=newnode();
if(v==-1) return 0;
if(x<=siz[ch[p][0]]) return _getkth(x,ch[p][0],v-1);
else return _getkth(x-siz[ch[p][0]],ch[p][1],v-1)|1<<v;
}
void insert(int x){_insert(x+1e7,rt,30);}
void remove(int x){_remove(x+1e7,rt,30);}
int getrank(int x){return _getrank(x+1e7,rt,30);}
int getkth(int x){return _getkth(x+1,rt,30)-1e7;}
int getpre(int x){return getkth(getrank(x)-1);}
int getsuf(int x){return getkth(getrank(x+1));}
};
int n;
trie<3100010> t;
int main(){
// freopen(".in","r",stdin);
// freopen(".out","w",stdout);
scanf("%d",&n);
for(int i=1;i<=n;i++){int op,x;
scanf("%d%d",&op,&x);
if(op==1) t.insert(x);
else if(op==2) t.remove(x);
else if(op==3) printf("%d\n",t.getrank(x));
else if(op==4) printf("%d\n",t.getkth(x));
else if(op==5) printf("%d\n",t.getkth(t.getrank(x)-1));
else if(op==6) printf("%d\n",t.getkth(t.getrank(x+1)));
}
return 0;
}
标签:ch,return,trie,siz,tot,int,01,newnode,模板
From: https://www.cnblogs.com/caijianhong/p/16863424.html