Xor Sum HDU - 4825

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