hdu 4825 Xor Sum(字典树)

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 <iostream>
#include <string.h>

using namespace std;

const int M = 55;
const int N = M*1e5;

struct Node
{
    long long value;
    int left, right;

    void clear()
    {
        left = right = -1;
    }
};
Node nodes[N];

int pid;    // 字典树数组Id分配
long long weight[N];

void insert(int &root, int d, long long value)
{
    if (root == -1)
    {
        root = pid++;
        nodes[root].clear();
    }

    if (d == -1)
    {
        nodes[root].value = value;
        return;
    }

    if ((weight[d] & value) != 0)
    {
        insert(nodes[root].right, d - 1, value);
    }
    else
    {
        insert(nodes[root].left, d - 1, value);
    }
}

void query(int root, int d, long long value)
{
    if (d == -1)
    {
        printf("%I64d\n", nodes[root].value);
        return;
    }

    if (((weight[d] & value) != 0) && (nodes[root].left != -1) || (nodes[root].right == -1))
    {
        query(nodes[root].left, d - 1, value);
    }
    else
    {
        query(nodes[root].right, d - 1, value);
    }
}

int main()
{
    weight[0] = 1;
    for (int i = 1; i < 55; i++)
    {
        weight[i] = 2 * weight[i - 1];
    }

    int T;
    scanf("%d", &T);

    for (int i = 1; i <= T; i++)
    {
        pid = 0;
        int root = -1;

        int m, n;
        scanf("%d%d", &m, &n);
        for (int j = 0; j < m; j++)
        {
            long long temp;
            scanf("%I64d", &temp);
            insert(root, 50, temp);
        }

        printf("Case #%d:\n", i);
        for (int j = 0; j < n; j++)
        {
            long long temp;
            scanf("%I64d", &temp);
            query(root, 50, temp);
        }
    }

    system("pause");
    return 0;
}


你可能感兴趣的:(hdu 4825 Xor Sum(字典树))