[Hdu4825]Xor Sum（01字典树）

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;
}