HDU-4825(数组维护字典树,更节省内存的写法)
程序员文章站
2022-06-04 12:46:58
...
指针动态分配内存:
数组分配内存:
所以还是用数组嘛,虽然有点抽象,但简单易操作,而且代码短
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
#define maxn 2000005
#define son 2
int ch[maxn][2],sz,val[maxn];//节点编号,字典树节点个数,值
void init()
{
sz=1;
memset(ch,0,sizeof(ch));
memset(val,0,sizeof(val));
}
void insert(int x)
{
int u=0;
bool a[32];
int xx=x;
for(int i=0;i<32;i++) a[i]=x&1,x>>=1;
for(int i=31;i>=0;i--){
if(!ch[u][a[i]]) ch[u][a[i]]=sz++;
u=ch[u][a[i]];
}
val[u]=xx;
}
int find(int x)
{
int u=0;
bool a[32];
int xx=x;
for(int i=0;i<32;i++) a[i]=x&1,x>>=1;
for(int i=31;i>=0;i--){
if(ch[u][!a[i]]) u=ch[u][!a[i]];
else u=ch[u][a[i]];
}
return val[u];
}
int main()
{
int T,c=0;
scanf("%d",&T);
for(int c=1;c<=T;c++){
init();
int n,m,x;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++){
scanf("%d",&x);
insert(x);
}
printf("Case #%d:\n",c);
for(int i=1;i<=m;i++){
scanf("%d",&x);
printf("%d\n",find(x));
}
}
return 0;
}