转自:https://blog.csdn.net/zuzhiang/article/details/79872805
01
字
典
树
主
要
用
于
解
决
求
异
或
最
值
的
问
题
。
01字典树主要用于解决求异或最值的问题。
01字典树主要用于解决求异或最值的问题。
1. 01字典树是一棵二叉树,每条节点的两条边分别表示二进制的某一位值是0还是1,
2. 将某个从根节点出发路径上边的值连起来就可以得到一个二进制串,这个二进制串是从根节点出发且最高位靠与根节点相连
3. 通过贪心的策略来寻找与x异或结果最大的数,即优先找和 x二进制的未处理的最高位值不同的边对应的点,这样保证结果最大
4. 这张图片讲的是在最大为4位二进制,3,4,5 是怎么插在01字典树上,以及val怎么存
配合下面模板和 https://cn.vjudge.net/problem/HDU-4825 请看这张图
模板
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]; //返回结果
}
例题
https://cn.vjudge.net/problem/HDU-4825
AC code
#include <bits/stdc++.h>
using namespace std;
#define maxn 100000
#define ll long long
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){
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;
ch[u][v]=tol++;
}
u=ch[u][v];
}
val[u]=x;
}
ll query(ll 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,n,m;
ll a;
scanf("%d",&T);
for(int s=1;s<=T;s++){
init();
scanf("%d %d", &n, &m);
for(int i=0;i<n;i++){
scanf("%lld", &a);
insert(a);
}
printf("Case #%d:\n", s);
for(int i=0;i<m;i++){
scanf("%lld", &a);
printf("%d\n", query(a));
}
}
}