A - Xor Sum (数组-字典树)

A - Xor Sum

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<set>
#include<cstdio>
#include<string>
#include<cstring>
#include<iostream>
#include<algorithm>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=1e5;
int trie[5000000][2];
int vis[5000000];
int tot,rt;
int er[40];
void Build_trie(int *er,int f)
{
    rt=0;
    for(int j=35; j>=0; j--)
    {
        int y=er[j];
        if(!trie[rt][y])
            trie[rt][y]=++tot;
        rt=trie[rt][y];
    }
    vis[rt]=f;
}
int Find_trie(int *er)
{
    rt=0;
    for(int j=35; j>=0; j--)
    {
        int y=!er[j];
        if(!trie[rt][y])
        {
            y=!y;
        }
        rt=trie[rt][y];
    }
    return vis[rt];
}
int main()
{
    int ca,cas=1;
    scanf("%d",&ca);
    while(ca--)
    {
        tot=0;
        mem(trie,0);
        mem(vis,0);
        int n,m,x,i,f;
        scanf("%d%d",&n,&m);
        while(n--)
        {
            i=0;
            mem(er,0);
            scanf("%d",&x);
            f=x;
            while(x)
            {
                er[i]=x%2;
                x/=2;
                i++;
            }
            Build_trie(er,f);
        }
        printf("Case #%d:\n",cas++);
        while(m--)
        {
            i=0;
            scanf("%d",&x);
            memset(er,0,sizeof(er));
            while(x)
            {
                er[i]=x%2;     //不能是(x+1)%2; 会WR;
                x/=2;
                i++;
            }
            printf("%d\n",Find_trie(er));
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/team39/article/details/83279434