codeforces 706D trie树或 muliset

最新推荐文章于 2025-01-20 21:44:42 发布

原创最新推荐文章于 2025-01-20 21:44:42 发布 · 532 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一种使用Trie树和Muliset数据结构来处理插入、删除和查询的操作方法。通过对比两种数据结构的实现方式，文章详细讲解了如何利用这些结构寻找与二进制数不同的位，以尽可能构成全为1的二进制数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：插入，删除，询问三个操作。

分析：显然trie树都支持这三个操作，但是代码比较长，用muliset更短小精悍。但是两者的本质都是一样，都是找与二进制不同的，尽量凑成全是1.

muliset

#include<bits/stdc++.h>
using namespace std;
multiset<int> dp;
bool check(int a,int b)
{
	auto it = dp.lower_bound(a);
	if(it == dp.end())	return false;
	if(*it < b)	return true;
	else	return false;
}
int solve(int x)
{
	int ans = 0;
	for(int i = 30; i >= 0 ; i --)
	{
		if(!((x>>i)&1))	ans |= 1<<i;
		if(!check(ans , ans + (1<<i)))	ans ^= 1<<i;
	}
	return ans^x;
}
int main(void)
{
 	int q,a;
 	char c;
 	cin>>q;
 	dp.insert(0);
 	while(q--)
 	{
 		cin>>c>>a;
 		if(c=='+')
 			dp.insert(a);
 		else if(c == '-')
 		{
 			auto it = dp.find(a);
 			dp.erase(it);
		}
		else	cout<<solve(a)<<endl;
	}
}

trie树

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 20;
struct Trie
{
	int ch[maxn<<5][2],val[maxn<<5],sz = 0;
	void inin(){
		sz = 1;memset(ch[0],0,sizeof(ch[0]));
	}
	void insert(int x,int v)
	{
		int u = 0;
		for(int i = 30 ; i >=0 ; i--)
		{
			int c = (x>>i) &1;
			if(!ch[u][c])
			{
				memset(ch[sz],0,sizeof(ch[sz]));
				val[sz] = 0;
				ch[u][c] = sz++;
			}
			u = ch[u][c];
			val[u] += v;
		}
	}
	int query(int x)
	{
		int ans = 0,u = 0;
		for(int i = 30 ; i >= 0; i --)
		{
			int id = (x >> i) & 1;
			if(!ch[u][id^1] || !val[ch[u][id^1]])
				u = ch[u][id];
			else
			{
				u = ch[u][id^1];
				ans |= 1<<i;
			}
		}
		return ans;
	}
}tree;
int main(void)
{
	tree.inin();
	int q,a;
 	char c;
 	cin>>q;
 	tree.insert(0,1);
 	while(q--)
 	{
 		cin>>c>>a;
 		if(c=='+')
 			tree.insert(a,1);
 		else if(c == '-')
 		{
 			tree.insert(a,-1);
		}
		else	cout<<tree.query(a)<<endl;
	}
}