HDU4825Xor 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

分析:
首先我们将n个数建立一个字典树,最大节点是100000*32,把数字转化为二进制建立字典树,查询时,我们考虑这样一个贪心,对于查询数字二进制的每一位,由于亦或的性质,我们想要字段数中的这一为和查询数中的这一位不相同,这样以后出的值在这一位是1

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <queue>
#include <set>
#include <vector>
using namespace std;
const int maxn = 100000*32+10;
struct tire
{
    int ch[maxn][5];
    long long val[maxn];
    int sz;
    void init()
    {
        sz=1;
        memset(ch[0],0,sizeof(ch[0]));
       // memset(val,0,sizeof(val));
    }
    void insert(int *s,long long v)
    {
        int u=0;
        for(int i=31;i>=0;i--)
        {
            int c=s[i];
            if(!ch[u][c])
            {
                memset(ch[sz],0,sizeof(ch[sz]));
                ch[u][c]=sz++;
                val[u]=0;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }
    long long find(int *s)
    {
        int u=0;
        for(int i=31;i>=0;i--)
        {
            int c=1-s[i];
            if(ch[u][c])
            {
                u=ch[u][c];
            }
            else
            {
                u=ch[u][s[i]];
            }
        }
        return val[u];
    }
}Tire;
int s[40];
int cnt;
void solve(long long x)
{
    cnt=0;
    while(x)
    {
        s[cnt++]=x%2; x/=2;
    }
    while(cnt<32)
    {
        s[cnt++]=0;
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    for(int kase=1;kase<=T;kase++)
    {
        long long n,m,x;
        scanf("%lld%lld",&n,&m);
        Tire.init();
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&x);
            solve(x);
            Tire.insert(s,x);
        }
        printf("Case #%d:\n",kase);
        for(int i=1;i<=m;i++)
        {
            scanf("%lld",&x);
            solve(x);
            long long ans=Tire.find(s);
            printf("%lld\n",ans);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/CURRYWANG97/article/details/81625911
今日推荐