hdu4825(Trie树+位运算)

链接:点击打开链接

题意:给出n个数和m次询问,每次询问给出一个数x,问在n个数中哪个数与x异或值最大

代码:

#include <queue>
#include <vector>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
int rt,a[1000005],v[3500005],s[3500005][2];
void in(int x,int id){
    int i,u,op;
    u=0;
    for(i=31;i>=0;i--){
        op=((x&(1<<i))!=0);
        if(s[u][op]==0)
        s[u][op]=rt++;
        u=s[u][op];
    }
    v[u]=id;
}
int get(int x){
    int i,u,op;
    u=0;
    for(i=31;i>=0;i--){
        op=((x&(1<<i))!=0);
        if(s[u][op^1]!=0)                       //尽可能走与当前位不同的点
        u=s[u][op^1];
        else
        u=s[u][op];
    }
    return a[v[u]];
}
int main(){                                     //将每个数拆分成二进制从高位到低位插入到trie树中,
    int t,n,m,i,j,u,cas;                        //每次询问都尽可能保证走不同与当前位不同的点,就
    scanf("%d",&t);                             //保证了异或值最大
    for(cas=1;cas<=t;cas++){
        scanf("%d%d",&n,&m);
        rt=1;
        memset(s,0,sizeof(s));
        memset(v,0,sizeof(v));
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
            in(a[i],i);
        }
        printf("Case #%d:\n",cas);
        for(i=1;i<=m;i++){
            scanf("%d",&u);
            printf("%d\n",get(u));
        }
    }
    return 0;
}

    原文作者:Trie树
    原文地址: https://blog.csdn.net/stay_accept/article/details/72847389
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞