HDU-4825 Xor Sum

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/C_13579/article/details/82664860

地址:http://acm.hdu.edu.cn/showproblem.php?pid=4825

思路:01字典树,对于查询的数x,查找树中的最大异或值,同时保存路径的数即可,还有在初始化时记得用num来处理

Code :

#include<iostream>
using namespace std;

const int MAX_S=3200005;
int n,Q,T;
int Trie[MAX_S][2],num;
bool Sum[MAX_S];

void Insert(int x);
int Query(int x);
int main()
{
	scanf("%d",&T);
	for(int t=1;t<=T;++t)
	{
		for(int i=0;i<=num;++i)
			Trie[i][0]=Trie[i][1]=Sum[i]=0;
		num=0;
		scanf("%d%d",&n,&Q);
		int x;
		for(int i=0;i<n;++i)
		{
			scanf("%d",&x);
			Insert(x);
		}
		printf("Case #%d:\n",t);
		while(Q--){
			scanf("%d",&x);
			printf("%d\n",Query(x));
		}
	}
	
	return 0;
}

void Insert(int x)
{
	int u=0,t;
	for(int i=31;i>=0;--i)
	{
		t=(x>>i)&1;
		if(!Trie[u][t])	Trie[u][t]=++num;
		u=Trie[u][t];
		Sum[u]=1;
	}
}

int Query(int x)
{
	int u=0,t,p,res=0;
	for(int i=31;i>=0;--i)
	{
		t=(x>>i)&1;
		p=Trie[u][!t];
		if(p&&Sum[p]){
			if(!t)	res=res|(1<<i);
			u=p;
		}else{
			if(t)	res=res|(1<<i);
			u=Trie[u][t];
		}
	}
	
	return res;
}

猜你喜欢

转载自blog.csdn.net/C_13579/article/details/82664860