百度之星2014资格赛 1003 - Xor Sum

先上代码：

Xor Sum

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

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树来保存集合里面的所有数，每个数在Trie树里面转化成一个32位的串，然后将查询的串根Trie数里面的串匹配，匹配原则是这样的：如果同一个位置有存在相异的位，那么就转向相异的位的方向(因为异或是同零异一)，如果不存在相异的位，那就只好走相同的位的方向了。这里可以分析得出一定存在路径是从头走到结尾的，因为这里每一个数都被扩展成32位，所以一定可以匹配到。

　　提交有点坑→_→，当时交上去都是WA和RE，赛后Rejudge有几次的提交过了，据说是卡输出，如果得到的结果用一个中间变量来保存的话会WA，如果直接输出结果的话才AC。目测赛后又改了，所以过了。

 

上代码：

 

#include <iostream>
#include <cstdio>
#include <cstring>
#include <bitset>
#define NOT(x) (x == 1 ? 0 : 1)
#define MAX 3200002
#define LL long long
using namespace std;

typedef bitset<32> bs;
int s[MAX][2],tot;

void reset(){
    memset(s[0],0,sizeof(s[0]));
    tot=1;
}

void insert(bs a){
    int j=0;
    for(int i=31;i>=0;i--){
        int u = a[i];
        if(!s[j][u]){
            memset(s[tot],0,sizeof(s[tot]));
            s[j][u] = tot++;
        }
        j = s[j][u];
    }
}

LL check(bs a){
    LL r=0;
    int j=0;
    for(int i=31;i>=0;i--){
        int u = a[i];
        r<<=1;
        if(s[j][NOT(u)]){
            r += NOT(u);
            j = s[j][NOT(u)];
        }else if(s[j][u]){
            r +=u;
            j = s[j][u];
        }

    }
    return r;
}

int main()
{
    LL t,n,m,a;
    LL r;
    //freopen("data.txt","r",stdin);
    scanf("%I64d",&t);
    for(int z=1;z<=t;z++){
        printf("Case #%d:\n",z);
        scanf("%I64d %I64d",&n,&m);
        reset();
        for(int i=0;i<n;i++){
            scanf("%I64d",&a);
            bs e(a);
            insert(e);
        }
        for(int i=0;i<m;i++){
            scanf("%I64d",&a);
            bs e(a);
            r=check(e);
            printf("%I64d\n",r);
        }
    }
    return 0;
}