HDU 4825+字典树

Xor Sum

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


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
 
  
23 23 4 5154 14 6 5 63
 

Sample Output
 
  
Case #1:43Case #2:4

//关键句:要在集合当中找出一个正整数 K ,使得 K 与 S 的异或结果最大
#include <cstdio>
#include <iostream>
#include <queue>
#include <algorithm>
#include <cstring>
#include <set>
#include <string>
#include <functional>
#include <vector>
#include <map>
#include <cmath>
using namespace std;
const int MAX = 2;
struct Trie {
    int num;
    bool flag;
    Trie* next[MAX];
    Trie() {
        num = 1;
        flag = false;
        memset(next, NULL, sizeof(next));
    }
};
Trie* NewTrie() {
    Trie* temp = new Trie;
    return temp;
}
void Insert(Trie* root, int S) {
    Trie* p = root;
    for (int i = 31; i >= 0; i--) {
        int id = ((S & (1 << i)) != 0);
        if (p->next[id] == NULL) {
            p->next[id] = NewTrie();
        }
        p = p->next[id];
    }
    p->flag = true;
}
int trie_query(Trie* root, int S) {
    Trie* p = root;
    int ans = 0;
    for (int i = 31; i >= 0; i--) {
        int id = ((S & (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;
}

int main() {

    int T, n, Q, t, ansk = 0;
    scanf("%d", &T);
    while (T--) {
        scanf("%d%d", &n, &Q);
        Trie* root = NewTrie();
        for (int i = 1; i <= n; i++){
            scanf("%d", &t);
            Insert(root, t);
        }
        printf("Case #%d:\n", ++ansk);
        while (Q--) {
            scanf("%d", &t);
            printf("%d\n", trie_query(root, t));
        }
    }
}

猜你喜欢

转载自blog.csdn.net/qq_34649947/article/details/79855372