HDU4825 Xor Sum【01字典树】

Xor Sum

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

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

 

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

Source

2014年百度之星程序设计大赛 - 资格赛

思路

01字典树求最大异或值。G++提交

C++代码

#include
#include

using namespace std;

const int N=100020;
int trie[32*N][2];
int val[32*N];

int tot;

void insert(int x)
{
	int rt=0;
	for(int i=31;i>=0;i--)
	{
		int c=(x>>i)&1;
		if(!trie[rt][c]) trie[rt][c]=++tot;
		rt=trie[rt][c];
	}
	val[rt]=x;
}

int find(int s)
{
	int rt=0;
	for(int i=31;i>=0;i--)
	{
		int c=(s>>i)&1;
		if(trie[rt][c^1])
		  rt=trie[rt][c^1];
		else
		  rt=trie[rt][c];
	}
	return val[rt];
}

int main()
{
	int T,n,m,s,x;
	scanf("%d",&T);
	for(int t=1;t<=T;t++)
	{
		tot=0;
		memset(val,0,sizeof(val));
		memset(trie,0,sizeof(trie));
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&x);
			insert(x);
		}
		printf("Case #%d:\n",t);
		while(m--)
		{
			scanf("%d",&s);
			printf("%d\n",find(s));
		} 
	}
	return 0;
}