HDU-4825-Xor Sum【字典树】

HDU-4825-Xor Sum【字典树】

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

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

题目链接：HDU-4825

题目大意：给出n长度的序列，m次询问,每次询问给出一个数字S，找出n序列中一个值K与S异或最大。

题目思路：很裸的字典树。

#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <string>
#include <cstring>
using namespace std;
#define MAXN 4000005 
struct Trie  // 定义01Trie 
{
    int next[2];
    int flag,number;
    void ini()  //初始化 
    {  
        next[0] = next[1] = 0;  
        flag = 0;  
    } 
}pnode[MAXN];
int root = 0;
int tot = 0;
void Insert(int num)
{
    int rt = root;
    for (int i = 30; i >= 0; i--)
    {
        int ret;
        if (num & (1 << i)) ret = 1; //得到第i位是1还是0 
        else ret = 0;
        if (!pnode[rt].next[ret]) //如果没有这个节点 
        {
            pnode[rt].next[ret] = ++tot;    //新增加这个节点，并且把这个子节点标记为tot 
            pnode[tot].number = ret;    //tot这个子节点的值为ret 
        }  
        rt = pnode[rt].next[ret];   // 获得子节点的编号 
        pnode[rt].flag++;          
    }
} 

int Query(int num)
{
    int rt = root;
    for (int i = 30; i >= 0; i--)
    {
        int ret;
        if (num & (1 << i)) ret = 1;
        else ret = 0;
        if (ret == 1)   //如果为1，子节点先搜0,0不存在再走1. 
        {
            int next_node = pnode[rt].next[0];
            if (pnode[next_node].flag > 0 && next_node) rt = pnode[rt].next[0];  //0存在
            else
            {
                rt = pnode[rt].next[1];
                num ^= (1 << i);
            }   
        }
        else  //如果为0，子节点先搜1,1不存在再走0. 
        {
            int next_node = pnode[rt].next[1];
            if (pnode[next_node].flag > 0 && next_node)  //1存在 
            {
                rt = pnode[rt].next[1];
                num ^= (1 << i);
            } 
            else rt = pnode[rt].next[0];
        }   
    }
    return num;
} 
int a[MAXN];
int cnt = 1;
int main(){
    int t;
    scanf("%d",&t);
    while(t--)
    {
        for(int i = 0;i < tot;i++)   //初始化 
        {  
            pnode[i].ini();  
        }
        int n,m;
        scanf("%d%d",&n,&m);
        for (int i = 0; i < n; i++)
        {
            scanf("%d",&a[i]);
            Insert(a[i]);
        }
        printf("Case #%d:\n",cnt++);
        while(m--)
        {
            int s;
            scanf("%d",&s);
            int ans = Query(s) ^ s;  //注意这里，异或得到K这个值
            printf("%d\n",ans); 
        }
    } 
    return 0;
}