【CF706D】Vasiliy's Multiset Trie+贪心

题目大意:需要维护一种数据结构,支持以下三种操作:插入一个数,删除一个数,查询该数据结构中的数异或给定数的最大值。

题解:如果没有删除操作就是一个标准的 Trie 上贪心求最大异或和问题。现在需要支持删除操作,因此,在树上每个节点维护一个额外的标记,表示有多少个数的某一位经过当前节点。插入操作依然只需修改树上一条链,而删除一个数时,同样需要将这条链上的标记值减 1 即可。这时便可以根据经过的每个点标记值是否为 0 进行贪心操作。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e5+1;

int trie[maxn<<5][2],tot=1,tag[maxn<<5];
int q,x;

void modify(int now,int idx,int val){
	if(idx<0)return;
	int ch=x>>idx&1;
	if(!trie[now][ch])trie[now][ch]=++tot;
	now=trie[now][ch],tag[now]+=val;
	modify(now,idx-1,val);
}

int query(int now,int idx,int ans){
	if(idx<0)return ans;
	int ch=x>>idx&1;
	if(tag[trie[now][ch^1]])now=trie[now][ch^1],ans|=1<<idx;
	else now=trie[now][ch];
	return query(now,idx-1,ans);
}

void solve(){
	char op[2];
	modify(1,31,1);
	while(q--){
		scanf("%s%d",op,&x);
		if(op[0]=='+')modify(1,31,1);
		else if(op[0]=='-')modify(1,31,-1);
		else printf("%d\n",query(1,31,0));
	}
}

int main(){
	scanf("%d",&q);
	solve();
	return 0;
}
posted @ 2018-12-18 23:24  shellpicker  阅读(183)  评论(0编辑  收藏  举报