hdu 4825 (Xor Sum) 字典树异或问题

版权声明： https://blog.csdn.net/weixin_39778570/article/details/82017442

题目链接

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

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

01静态字典树
使用数组实现

AC code

#define _CRT_SBCURE_NO_DEPRECATE
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>

using namespace std;

#define maxn 4000010  // 注意大小要开大点，因为这个吃了很多个TEL，RT
#define maxm 2
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1

int trie[maxn][maxm] = {0};
int t,n,m,a,ans,k=1, b[32];

void insert(int x){
    for(int i=31; i>=0; i--){
        b[i] = x&1;
        x >>= 1;
    }
    int p = 0;
    for(int i=0; i<32; i++){
        int c = b[i];
        if(!trie[p][c]){
            trie[p][c] = k;
            k++;
        }
        p = trie[p][c];
    }
}

int search(int x){
    int ret = 0;
    for(int i=31; i>=0; i--){
        b[i] = x&1;
        x >>= 1;
    }
    int p=0;
    for(int i=0; i<32; i++){
        int c = 1-b[i];
        if(!trie[p][c]){
            p = trie[p][1-c];
            ret = ret << 1;
        }else{
            p = trie[p][c]; // 至少有一个数所以可以匹配到
            ret = ret<<1|1;
        }
    }
    return ret;
}

int main(){
    scanf("%d", &t);
    int i=1;
    while(t--){
        memset(trie, 0, sizeof(trie));
        scanf("%d%d", &n, &m);
        for(int i=0; i<n; i++){
            scanf("%d", &a);
            insert(a);
        }
        printf("Case #%d:\n", i++); 
        for(int i=0; i<m; i++){
            scanf("%d", &a);
            ans = search(a); // 最大结果
            ans ^= a;        // 取回数组集里的数
            printf("%d\n", ans);
        } 
    }
}

使用结构体

#define _CRT_SBCURE_NO_DEPRECATE
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>

#define maxn 100005
#define maxm 2
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1

using namespace std;
typedef long long ll;

struct node{
    ll next[2];
};

ll t,n,m,k;
//node trie[100005];
node trie[4000010];
void insert(ll now, ll num){
    for(int i=32; i>=0; --i){  // 第32为为0(不存在)，存在1号根节点里 
        ll v = ((num>>i) & 1);  // 取出第i位
        if(trie[now].next[v] == -1){
            trie[now].next[v] = ++k; // 创建节点 
        } 
        now = trie[now].next[v];
    }
}

ll search(ll now, ll num){
    ll re = 0;
    for(ll i=32; i>=0; --i){
        ll v = ((num>>i) & 1);
        v = !v; // 取反
        if(trie[now].next[v] != -1){
            re = (re<<1);
            re = (re | v); // 取数字集里的数 
            now = trie[now].next[v];
        }else{
            re = (re<<1);
            v = !v;  // 取原数 
            re = (re|v);
            now = trie[now].next[v]; // 一定有路    
        }
    }   
    return re;
}

int main(){
    ll ac = 0;
    scanf("%d", &t);
    while(t--){
        ++ac;
        scanf("%I64d%I64d",&n,&m);
        k = 1;
        memset(trie, -1, sizeof(trie));
        for(ll i=1; i<=n; i++){
            ll num;
            scanf("%I64d", &num);
            insert(1, num);
        }
        printf("Case #%I64d:\n",ac);
        for(ll i=1; i<=m; i++){
            ll num, ans;
            scanf("%I64d", &num);
            ans = search(1, num);
            printf("%I64d\n", ans);
        }
    }   
    return 0;
}