HDU - 4825 Xor Sum 01字典树裸题

题目链接

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

思路：01字典树

#include<iostream>
#include<iomanip>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#include<queue>
#include<stack>
#include<vector>
#include<map>
#include<set>
#include<cstdlib>
#define rep(i,a,n) for(int i=a;i<=n;i++)
#define ll long long
#define ull unsigned long long
#define mem(a,x) memset((a),(x),sizeof ((a)))//x只能是0或-1或false或true
#define debug(x) cout<<"X: "<<(x)<<endl
#define de cout<<"************"<<endl
#define lowbit(x) ((x)&(-x))
#define lson rt<<1
#define rson rt<<1|1
#define gcd(a,b) __gcd(a,b)
#define lcm(a,b) a*b/(__gcd(a,b))
#define inf 0x3f3f3f3f//1e9+6e7
#define eps 1e-8
#define mod 1e9+7
#define N 100010
const double pi=acos(-1.0);
using namespace std;
ll tol; //节点个数
ll val[32*N]; //点的值
ll trie[32*N][2]; //边的值

void init() //初始化
{
    tol=1;
    trie[0][0]=trie[0][1]=0;
    mem(trie,0);
    mem(val,inf);
}

void insertt(ll x)
{
    //往 01字典树中插入 x
    int u=0;
    for(int i=31; i>=0; i--)
    {
        int v=(x>>i)&1;
        if(!trie[u][v])
        {
            //如果节点未被访问过
            trie[u][v]=tol++; //边指向的节点编号
        }
        u=trie[u][v]; //下一节点
    }
    val[u]=x; //节点值为 x，即到此是一个数
}

ll query(ll x)
{
    //查询所有数中和 x异或结果最大的数,返回那个数
    int u=0;
    for(int i=31; i>=0; i--)
    {
        int v=(x>>i)&1;
        //利用贪心策略，优先寻找和当前位不同的数
        if(trie[u][v^1]) u=trie[u][v^1];
        else    u=trie[u][v];
    }
    return val[u]; //返回结果
}

int main()
{
    int T;
    cin>>T;
    int tt=0;
    while(T--)
    {
        tt++;
        init();
        int n,m;
        cin>>n>>m;
        ll x;
        rep(i,1,n)
        {
            scanf("%lld",&x);
            insertt(x);
        }
        cout<<"Case #"<<tt<<":"<<endl;
        rep(i,1,m)
        {
            scanf("%lld",&x);
            cout<<query(x)<<endl;
            //cout<<(query(x)^x)<<endl;
        }
    }
    return 0;
}