Hdu4825Xor Sum（01字典树）

Posted 2020-10-27 void_f

Description

Zeus 和 Prometheus 做了一个游戏，Prometheus 给 Zeus 一个集合，集合中包含了N个正整数，随后 Prometheus 将向 Zeus 发起M次询问，每次询问中包含一个正整数 S ，之后 Zeus 需要在集合当中找出一个正整数 K ，使得 K 与 S 的异或结果最大。Prometheus 为了让 Zeus 看到人类的伟大，随即同意 Zeus 可以向人类求助。你能证明人类的智慧么？

Solution

01字典树入门题

Code

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

int T,n,m,ch[3000010][2],rt,v[3000010],A[100010];

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void Insert(int x,int id){
    int u=0,op;
    for(int i=31;i>=0;--i){
        op=((x&(1<<i))!=0);
        if(!ch[u][op]) ch[u][op]=rt++;
        u=ch[u][op];
    }
    v[u]=id;
}

int get(int x){
    int u=0,op;
    for(int i=31;i>=0;--i){
        op=((x&(1<<i))!=0);
        if(ch[u][op^1]!=0) u=ch[u][op^1];
        else u=ch[u][op];
    }
    return A[v[u]]; 
}

int main(){
    T=read();
    for(int cas=1;cas<=T;++cas){
        n=read(),m=read();
        rt=1;
        memset(ch,0,sizeof(ch));
        memset(v,0,sizeof(v));
        for(int i=1;i<=n;++i) Insert(A[i]=read(),i);
        printf("Case #%d:\n",cas);
        while(m--) printf("%d\n",get(read()));
    }
    return 0;
}