Xor Sum HDU - 4825

最新推荐文章于 2020-08-26 17:18:11 发布

sunyutian1998

最新推荐文章于 2020-08-26 17:18:11 发布

阅读量214

点赞数

CC 4.0 BY-SA版权

分类专栏： 01字典树文章标签： 01字典树 HDU

本文链接：https://blog.youkuaiyun.com/sunyutian1998/article/details/82589145

01字典树专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一道经典的字典树模板题，通过数组和指针两种方式实现字典树来解决该问题。文章详细解释了如何利用字典树进行更新和查询操作，并提供了完整的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

01字典树模板题

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

数组版

#include <bits/stdc++.h>
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 <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;
}