【模板】01-trie

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;
}