[01字典树] 询问异或最大的数 HDU4825

Xor Sum

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

 

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年百度之星程序设计大赛 - 资格赛

 

#include <bits/stdc++.h>
using namespace std;

const int mn = 100010;
int a[mn];

int num;
int trie[mn * 35][2];
int ed[mn * 35];
void add(int t, int id)
{
	int ch[35] = {0};
	int len = 0;
	while (t > 0)
	{
		ch[++len] = t & 1;
		t >>= 1;
	}
	int r = 0;
	for (int i = 32; i > len; i--)  // 前补0补齐32位
	{
		if (trie[r][0] == 0)
			trie[r][0] = ++num;
		r = trie[r][0];
	}
	for (int i = len; i > 0; i--)
	{
		if (trie[r][ch[i]] == 0)
			trie[r][ch[i]] = ++num;
		r = trie[r][ch[i]];
	}
	ed[r] = id;
}
int ask(int t)
{
	int ch[35] = {0};
	int len = 0;
	while (t > 0)
	{
		ch[++len] = t & 1;
		t >>= 1;
	}
	int r = 0;
	for (int i = 32; i > len; i--)
	{
		if (trie[r][1] != 0)
			r = trie[r][1];
		else
			r = trie[r][0];
	} // 高位异或不同沿高位向下
	for (int i = len; i > 0; i--)
	{
		if (trie[r][ch[i] ^ 1] != 0)
			r = trie[r][ch[i] ^ 1];
		else
			r = trie[r][ch[i]];
	}
	return ed[r];
}

int main()
{
	int T;
	scanf("%d", &T);
	for (int cas = 1; cas <= T; cas++)
	{
		num = 0;
		memset(ed, 0, sizeof ed);
		memset(trie, 0, sizeof trie);
		printf("Case #%d:\n", cas);
		
		int n, m;
		scanf("%d %d", &n, &m);
		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &a[i]);
			add(a[i], i);
		}
		while (m--)
		{
			int p;
			scanf("%d", &p);
			int t = ask(p);
			printf("%d\n", a[t]);
		}
	}
}