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


原文地址:http://www.cnblogs.com/caijianhong/p/16863424.html

1. 本站所有资源来源于用户上传和网络,如有侵权请邮件联系站长! 2. 分享目的仅供大家学习和交流,请务用于商业用途! 3. 如果你也有好源码或者教程,可以到用户中心发布,分享有积分奖励和额外收入! 4. 本站提供的源码、模板、插件等等其他资源,都不包含技术服务请大家谅解! 5. 如有链接无法下载、失效或广告,请联系管理员处理! 6. 本站资源售价只是赞助,收取费用仅维持本站的日常运营所需! 7. 如遇到加密压缩包,默认解压密码为"gltf",如遇到无法解压的请联系管理员! 8. 因为资源和程序源码均为可复制品,所以不支持任何理由的退款兑现,请斟酌后支付下载 声明:如果标题没有注明"已测试"或者"测试可用"等字样的资源源码均未经过站长测试.特别注意没有标注的源码不保证任何可用性