【YbtOJ算法高效进阶2023 「字符串算法」第4章 字典树 #B】最大异或对(字典树)

题目大意

给定的 $N$ 个整数 $A_i$，选出两个数进行 $xor$（异或）运算，求得到的结果的最大值。

输入格式
第一行输入一个整数 $N$。
第二行输入 $N$ 个整数 $A_i$。

输出格式
输出一个整数表示答案。

输入样例

3 
1 2 3

输出样例

3

基本思路

这题好办，我们把每个$A_i$转成二进制放进一个$trie$树里就可以了，这样我们选中一个数让它尝试和别的数异或，在$trie$树上我们每次尽量往相反的方向走，比如这个数在二进制下第一位是$1$，我们就看看有没有二进制下第一位是$0$的，然后往那个方向走就行了。如此枚举即可。

核心代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int A=1e5*31+10;
const int N=1e5+10;
int trie[A][3],tot=1;
int n,ans,a[N];
inline void insert(int x){
	int p=1;
	//转化为二进制储存,统一为31位,方便比较最高位 
	for(int i=31;i>=0;i--){
		int cnt=(x>>i)&1;
		if(!trie[p][cnt]) trie[p][cnt]=++tot;
		p=trie[p][cnt];
	}
	//都是31位,就没必要标记结束位置了 
}
inline int query(int x){//查询一个数为x的最大异或值 
	int p=1,nans=0;
	for(int i=31;i>=0;i--){
		int cnt=((x>>i)&1)==1?0:1;
		//如果是1,cnt=0,反之cnt=1
		//我们希望高位不同的越多越好,故尽量往不同的
		//方向走
		if(trie[p][cnt]) nans+=(int)(1<<i);//加入答案
		else cnt=((x>>i)&1);//相同异或为0
		p=trie[p][cnt]; 
	}
	return nans;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&(a[i]));
		insert(a[i]);
	}
	for(int i=1;i<=n;i++) ans=max(ans,query(a[i]));
	printf("%d",ans);
	return 0;
}