【Trie树求最大异或】HDU - 4825 - Xor Sum

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/monochrome00/article/details/81782553

题目链接<http://acm.hdu.edu.cn/showproblem.php?pid=4825>


题意:

有一个集合,里面含有n个数。有m次询问,每次询问一个数和集合里面的数异或,使得到的值最大,应该和哪个数异或。


题解:

利用Trie树,将n个数的32位二进制从高位开始放进树中。将询问的数字在树里贪心的查询,这样复杂度是n*32。l


#include<iostream>
#include<cstdio>
#include<cmath>
#include<queue>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const ll N=5e6+7;
ll nxt[N][2],val[N],rt,cnt,n,m;
ll a[N];
void insert(ll x,ll id){
    ll now=rt;
    for(ll i=32;i>=0;i--){
        ll w=x&((ll)1<<i); if(w>1) w=1;
        if(!nxt[now][w]) nxt[now][w]=++cnt;
        now=nxt[now][w];
    }
    val[now]=a[id];
}
ll query(ll x){
    ll now=rt;
    for(ll i=32;i>=0;i--){
        ll w=x&((ll)1<<i); if(w>1) w=1;
        if(!nxt[now][!w]) now=nxt[now][w];
        else now=nxt[now][!w];
    }
    return val[now];

}
int main(){
    ll T;
    scanf("%lld",&T);
    for(ll t=1;t<=T;t++){
        memset(nxt,0,sizeof(nxt));
        memset(val,0,sizeof(val));
        scanf("%lld%lld",&n,&m);
        for(ll i=1;i<=n;i++){
            scanf("%lld",&a[i]);
            insert(a[i],i);
        }
        printf("Case #%lld:\n",t);
        ll x;
        for(ll i=1;i<=m;i++){
            scanf("%lld",&x);
            ll k=query(x);
            printf("%lld\n",k);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/monochrome00/article/details/81782553
今日推荐