Xor Sum HDU - 4825 01字典樹模板

題意:n個數m次查詢,問你每次查詢n個數裏面的哪個數與x異或值最大

題解:其實就是每次看每一位儘量跟x不一樣的串,貪心思想。

馬上要用01字典樹了,先點一下技能點

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define mem(s) memset(s, 0, sizeof(s))
const int INF = 0x3f3f3f3f;
const double eps = 1e-8;
const int maxn = 1e6+5;
const int mod = 998244353;
int tol; //節點個數 
ll val[32*maxn]; //點的值 
int ch[32*maxn][2]; //邊的值 
 
void init()
{ //初始化 
    tol=1;
    ch[0][0]=ch[0][1]=0;
}
void insert(ll x)
{ //往 01字典樹中插入 x 
    int u=0;
    for(int i=32;i>=0;i--)
    {
        int v=(x>>i)&1;
        if(!ch[u][v])
        { //如果節點未被訪問過 
            ch[tol][0]=ch[tol][1]=0; //將當前節點的邊值初始化 
            val[tol]=0; //節點值爲0,表示到此不是一個數 
            ch[u][v]=tol++; //邊指向的節點編號 
        }
        u=ch[u][v]; //下一節點 
    }
    val[u]=x; //節點值爲 x,即到此是一個數 
}
 
ll query(ll x)
{ //查詢所有數中和 x異或結果最大的數 
    int u=0;
    for(int i=32;i>=0;i--)
    {
        int v=(x>>i)&1;
        //利用貪心策略,優先尋找和當前位不同的數 
        if(ch[u][v^1]) u=ch[u][v^1];
        else u=ch[u][v];
    }
    return val[u]; //返回結果 
}
int main() 
{
    int t,kase=0;
    scanf("%d",&t);
    while(t--){
        init();
        int n,m;
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            ll tmp;
            scanf("%lld",&tmp);
            insert(tmp);
        }
        vector<ll>ans;
        for(int i=1;i<=m;i++){
            int tmp;
            scanf("%d",&tmp);
            ans.push_back(query(tmp));
            //printf("%lld\n",ans);
        }
        printf("Case #%d:\n",++kase);
        for(int i=0;i<ans.size();i++){
            printf("%lld\n",ans[i]);
        }
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章