题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=4825
题目:
n个数,q个询问,每个询问给出一个数x, 输出这n个数中与x异或值最大的那个数
题解&笔记:
01字典树模版题。
01字典树解决从一堆数中选择一个数和给定的数x异或值最大的问题。
字典树上存的是每一位的值,0或者1,每个节点的下一层都有两个节点0/1,用数组val标记数值节点。
在寻找和x异或值最大的那个数, 每次都在01字典树上贪心的找和当前位异或起来是1的那个节点,最终找到数值节点。
ac代码:
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=100005;// 8!=40320!!不是10000
const int max_base = 32;
ll val[32*maxn];
int ch[32*maxn][2];
int tol = 0;
void init()
{
tol = 1;
ch[0][0] = ch[1][0] = 0;
}
void insert(ll x)
{
int u = 0;
for(int i = max_base; i >= 0; i--)
{
int v = (x >> i & 1);
if(!ch[u][v])
{
ch[tol][1] = ch[tol][0] = 0;
val[tol] = 0; //不是数值节点
ch[u][v] = tol++;
}
u = ch[u][v];
}
val[u] = x;
}
ll query(ll x)
{
int u = 0;
for(int i = max_base; i >= 0; i--)
{
int v = x >> i & 1;
if(ch[u][v^1]) u = ch[u][v^1]; // 贪心的找每位异或起来的1的节点
else u = ch[u][v];
}
return val[u];
}
int main()
{
//freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
int t;
while(~scanf("%d", &t))
{
int n, q;
ll x;
for(int k = 1; k <= t; k++)
{
printf("Case #%d:\n", k);
init();
scanf("%d %d", &n, &q);
for (int i = 0; i < n; i++)
{
scanf("%lld", &x);
insert(x);
}
for (int i = 0; i < q; i++)
{
scanf("%lld", &x);
printf("%lld\n", query(x));
}
}
}
return 0;
}