Xor Sum HDU - 4825

http://acm.hdu.edu.cn/showproblem.php?pid=4825

01字典树模板题

贪心的考虑 肯定要先使最高位为1 在这基础上递归考虑低位 所以插入和查询时都是高位开头

 

数组版

#include 
using namespace std;
#define ll long long

struct node
{
    int c[2];
    int cnt;
};

node tree[3000010];
ll pre[50];
int n,q,num;

void init()
{
    int i;
    pre[0]=1;
    for(i=1;i<=31;i++) pre[i]=2ll*pre[i-1];
}

void update(int val)
{
    int cur,i,id;
    cur=0;
    for(i=31;i>=0;i--)
    {
        id=1&(val>>i);
        if(tree[cur].c[id]==0) tree[cur].c[id]=num++;
        tree[tree[cur].c[id]].cnt++;
        cur=tree[cur].c[id];
    }
}

ll query(ll val)
{
    ll res;
    int cur,i,id;
    cur=0,res=0;
    for(i=31;i>=0;i--)
    {
        id=1&(val>>i);
        if(tree[cur].c[id^1])
        {
            cur=tree[cur].c[id^1];
            res+=(id^1)*pre[i];
        }
        else
        {
            cur=tree[cur].c[id];
            res+=id*pre[i];
        }
    }
    return res;
}

int main()
{
    ll val;
    int t,cas,i;
    init();
    scanf("%d",&t);
    for(cas=1;cas<=t;cas++)
    {
        scanf("%d%d",&n,&q);
        memset(tree,0,sizeof(tree));
        num=1;
        for(i=1;i<=n;i++)
        {
            scanf("%lld",&val);
            update(val);
        }
        printf("Case #%d:\n",cas);
        while(q--)
        {
            scanf("%lld",&val);
            printf("%lld\n",query(val));
        }
    }
    return 0;
}

 

指针版

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