http://acm.hdu.edu.cn/showproblem.php?pid=4825
01字典树模板题
贪心的考虑 肯定要先使最高位为1 在这基础上递归考虑低位 所以插入和查询时都是高位开头
#include <bits/stdc++.h>
using namespace std;
#define ll long long
struct node
{
node *next[2];
bool cnt;
};
node *root;
int n,q;
void getnode(node *&ptr)
{
int i;
ptr=new node;
for(i=0;i<2;i++) ptr->next[i]=NULL;
ptr->cnt=0;
}
void update(node *ptr,int *ary,int cur,int len)
{
if(ptr->next[ary[cur]]==NULL) getnode(ptr->next[ary[cur]]);
if(cur==len-1)
{
ptr->next[ary[cur]]->cnt=1;
return;
}
update(ptr->next[ary[cur]],ary,cur+1,len);
}
ll query(node *ptr,int *ary,ll val,int cur,int len)
{
if(cur==len) return 0;
if(ptr->next[ary[cur]^1]!=NULL) return val*(ary[cur]^1)+query(ptr->next[ary[cur]^1],ary,val/2,cur+1,len);
else return val*ary[cur]+query(ptr->next[ary[cur]],ary,val/2,cur+1,len);
}
int main()
{
ll val;
int ary[50];
int t,cas,i;
scanf("%d",&t);
for(cas=1;cas<=t;cas++)
{
getnode(root);
scanf("%d%d",&n,&q);
while(n--)
{
scanf("%lld",&val);
for(i=34;i>=0;i--,val/=2) ary[i]=val%2;
update(root,ary,0,35);
}
printf("Case #%d:\n",cas);
while(q--)
{
scanf("%lld",&val);
for(i=34;i>=0;i--,val/=2) ary[i]=val%2;
printf("%lld\n",query(root,ary,17179869184ll,0,35));
}
}
return 0;
}