hdoj 4825 Xor Sum 【字典树】

Xor Sum

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 132768/132768 K (Java/Others)

Total Submission(s): 1020    Accepted Submission(s): 411

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就OK了。

AC代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <vector>
#include <string>
#define INF 0x3f3f3f3f
#define eps 1e-8
#define MAXN (4000000+10)
#define MAXM (200000+10)
#define Ri(a) scanf("%d", &a)
#define Rl(a) scanf("%lld", &a)
#define Rf(a) scanf("%lf", &a)
#define Rs(a) scanf("%s", a)
#define Pi(a) printf("%d\n", (a))
#define Pf(a) printf("%.2lf\n", (a))
#define Pl(a) printf("%lld\n", (a))
#define Ps(a) printf("%s\n", (a))
#define W(a) while(a--)
#define CLR(a, b) memset(a, (b), sizeof(a))
#define MOD 1000000007
#define LL long long
#define lson o<<1, l, mid
#define rson o<<1|1, mid+1, r
#define ll o<<1
#define rr o<<1|1
#define PI acos(-1.0)
using namespace std;
struct Trie{
int next[MAXN][2], root, L, word[MAXN];
int newnode()
{
for(int i = 0; i < 2; i++)
next[L][i] = -1;
word[L++] = 0;
return L-1;
}
void init()
{
L = 0;
root = newnode();
}
void Insert(int val)
{
int u = root, v;
for(int i = 30; i >= 0; i--)
{
v = (val & (1<<i)) ? 1 : 0;
if(next[u][v] == -1)
next[u][v] = newnode();
u = next[u][v];
word[u]++;
}
}
void Delete(int val)
{
int u = root, v;
for(int i = 30; i >= 0; i--)
{
v = (val & (1<<i)) ? 1 : 0;
u = next[u][v];
word[u]--;
}
}
int Query(int val)
{
int u = root, v;
int ans = 0;
for(int i = 30; i >= 0; i--)
{
v = (val & (1<<i)) ? 1 : 0;
if(v == 1)
{
if(next[u][0] != -1 && word[next[u][0]])
u = next[u][0];
else
u = next[u][1], ans |= (1 << i);
}
else
{
if(next[u][1] != -1 && word[next[u][1]])
u = next[u][1], ans |= (1 << i);
else
u = next[u][0];
}
}
return ans;
}
};
Trie solve;
int main()
{
int t, kcase = 1; Ri(t);
W(t)
{
int N, Q, K, a; Ri(N); Ri(Q);
solve.init();
W(N) scanf("%d", &a), solve.Insert(a);
printf("Case #%d:\n", kcase++);
W(Q) Ri(K), Pi(solve.Query(K));
}
return 0;
}