码迷,mamicode.com
首页 > 其他好文 > 详细

【Hdu4825】Xor Sum(01字典树)

时间:2018-03-14 21:10:24      阅读:117      评论:0      收藏:0      [点我收藏+]

标签:line   字典   using   str   getchar   --   oid   ++i   集合   

Description

Zeus 和 Prometheus 做了一个游戏,Prometheus 给 Zeus 一个集合,集合中包含了N个正整数,随后 Prometheus 将向 Zeus 发起M次询问,每次询问中包含一个正整数 S ,之后 Zeus 需要在集合当中找出一个正整数 K ,使得 K 与 S 的异或结果最大。Prometheus 为了让 Zeus 看到人类的伟大,随即同意 Zeus 可以向人类求助。你能证明人类的智慧么?

Solution

01字典树入门题

Code

#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

int T,n,m,ch[3000010][2],rt,v[3000010],A[100010];

inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void Insert(int x,int id){
    int u=0,op;
    for(int i=31;i>=0;--i){
        op=((x&(1<<i))!=0);
        if(!ch[u][op]) ch[u][op]=rt++;
        u=ch[u][op];
    }
    v[u]=id;
}

int get(int x){
    int u=0,op;
    for(int i=31;i>=0;--i){
        op=((x&(1<<i))!=0);
        if(ch[u][op^1]!=0) u=ch[u][op^1];
        else u=ch[u][op];
    }
    return A[v[u]]; 
}

int main(){
    T=read();
    for(int cas=1;cas<=T;++cas){
        n=read(),m=read();
        rt=1;
        memset(ch,0,sizeof(ch));
        memset(v,0,sizeof(v));
        for(int i=1;i<=n;++i) Insert(A[i]=read(),i);
        printf("Case #%d:\n",cas);
        while(m--) printf("%d\n",get(read()));
    }
    return 0;
}

【Hdu4825】Xor Sum(01字典树)

标签:line   字典   using   str   getchar   --   oid   ++i   集合   

原文地址:https://www.cnblogs.com/void-f/p/8570063.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!