【Trie】hdu 4825 Xor Sum

Xor Sum

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 132768/132768 K (Java/Others)
Total Submission(s): 10482    Accepted Submission(s): 4323

Problem Description

Zeus 和 Prometheus 做了一个游戏，Prometheus 给 Zeus 一个集合，集合中包含了N个正整数，随后 Prometheus 将向 Zeus 发起M次询问，每次询问中包含一个正整数 S ，之后 Zeus 需要在集合当中找出一个正整数 K ，使得 K 与 S 的异或结果最大。Prometheus 为了让 Zeus 看到人类的伟大，随即同意 Zeus 可以向人类求助。你能证明人类的智慧么？

 

 

Input

输入包含若干组测试数据，每组测试数据包含若干行。
输入的第一行是一个整数T（T < 10），表示共有T组数据。
每组数据的第一行输入两个正整数N，M（<1=N,M<=100000），接下来一行，包含N个正整数，代表 Zeus 的获得的集合，之后M行，每行一个正整数S，代表 Prometheus 询问的正整数。所有正整数均不超过2^32。

 

 

Output

对于每组数据，首先需要输出单独一行”Case #?:”，其中问号处应填入当前的数据组数，组数从1开始计算。
对于每个询问，输出一个正整数K，使得K与S异或值最大。

 

 

Sample Input

2 3 2 3 4 5 1 5 4 1 4 6 5 6 3

 

 

Sample Output

Case #1: 4 3 Case #2: 4

 

把每个数字拆成二进制，加入到Trie树中

由于前面的位置比后面的影响更大，所以先确定前面的位置更优

 

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=4e6+5;
int ch[maxn][2],cnt[maxn];
int n,m,a[40],tot;
void divide(int x)
{
	memset(a,0,sizeof(a));
	for(int i=33;i>=0,x>0;i--,x>>=1)
		a[i]=x%2;
}
void insert(int x)
{
	divide(x);
	int now=0;
	for(int i=1;i<=33;i++)
	{
		if(!ch[now][a[i]]) ch[now][a[i]]=++tot;
		now=ch[now][a[i]];
	}
	cnt[now]=x;
}
int query(int x)
{
	divide(x);
	int now=0,res=0;
	for(int i=1;i<=33;i++)
	{
		if(!ch[now][!a[i]]) now=ch[now][a[i]];
		else now=ch[now][!a[i]];
		if(cnt[now]) res=max(res,cnt[now]);
	}
	return res;
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("a.out","w",stdout);
	int t,x; scanf("%d",&t);
	for(int k=1;k<=t;k++)
	{
		memset(ch,0,sizeof(ch));
		memset(cnt,0,sizeof(cnt));
		tot=0; scanf("%d%d",&n,&m);
		printf("Case #%d:\n",k);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&x);
			insert(x);	
		}
		for(int i=1;i<=m;i++)
		{
			scanf("%d",&x);
			printf("%d\n",query(x));
		}
	}
	return 0;
}