POJ 4825 Xor Sum 字典树

Xor Sum

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

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
   
   
以每个数高位在前地位在后构造字典树，查询时候匹配求最大值。
   
   

   
   
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxm = 100005;
int a[maxm], m;
typedef struct Trie
{
	Trie *next[2];
	Trie()
	{
		next[0] = next[1] = NULL;
	}
};
Trie *root;
void insert(int k);
int find(int k);
int main()
{
	int n, i, j, k, sum, x, t, cas = 0;
	scanf("%d", &t);
	while (t--)
	{
		printf("Case #%d:\n", ++cas);
		scanf("%d%d", &n, &m);
		root = new Trie;
		for (i = 1;i <= n;i++)
		{
			scanf("%d", &x);
			insert(x);
		}
		for (i = 1;i <= m;i++)
		{
			scanf("%d", &x);
			printf("%d\n", find(x));
		}
	}
	return 0;
}
void insert(int k)
{
	Trie *p = root, *q;
	for (int i = 31;i >= 0;i--)
	{
		int id = (k&(1 << i)) != 0;
		if (p->next[id] == NULL)
		{
			q = new Trie;
			p->next[id] = q;
			p = p->next[id];
		}
		else
			p = p->next[id];
	}
}
int find(int k)
{
	int ans = 0;
	Trie *p = root, *q;
	for (int i = 31;i >= 0;i--)
	{
		int id = (k&(1 << i)) != 0;
		if (p->next[id ^ 1] != NULL)
		{
			ans |= (id ^ 1) << i;
			p = p->next[id ^ 1];
		}
		else
		{
			ans |= id << i;
			p = p->next[id];
		}
	}
	return ans;
}