题目描述
自己看题了 最大异或对
能来看我题解的,想必都看过Y总的课程了。
我就说说我对这个Trie树算法解的两个问题
1、son[n][m]怎么理解,位移数位个数为什么是从30 - 0右移?
2、Trie树的构建逻辑
3、Trie的查询逻辑
算法
1、 son数组定义是二维数组,son[n][m]我在初学Trie树的时候很难理解,可以先理解它的第二维度,只有两种状态0/1,是因为这一位表示的是某个数字的的某一位是0 / 1,而第一维的大小是我们用的十足二进制表示下一共有多少位数,在本题中,数字一共有N个数字,N是小于$10^5$的,一个数在int下是32位,则我们需要至少3200000的一维坐标,而p = son[n][0] / p = son[n][1]其实存的值就是他的两个子节点的一维坐标的值。
那么x >> i & 1 其实就是我想知道x的二进制表示中的第i位(二进制位从第0位开始表示第0位 - 第 31 位),本题的题目范围到2^31那么就是i从30 - 0。
2、构建逻辑其实相对简单,就是将数的二进制表示插入到son数组中,如果没有那么就将他的值附上++idx即可
3、查询的时候为了异或大小最大,那么就应该从根节点开始遍历最高位开始尽量让异或结果为1,在遍历查询数字的第i位的时候取第i位的值为s,查看son[p][!s]是否存在,如果存在那么将res+= i << i(这一位异或存在最大值为1),如果没有那么就让p = son[p]s。
记得点赞啊!
时间复杂度 $O(nlogn)$
参考文献
Java 代码
import java.util.*;
import java.io.*;
class Main{
static int idx;
static int[][] son = new int[3000000][2];
static void insert(int x){
int p = 0;
for(int i = 30; i >= 0; i--){
if(son[p][(x >> i) & 1] == 0){
son[p][(x >> i) & 1] = ++ idx;
}
p = son[p][(x >> i) & 1];
}
}
static int query(int x){
int p = 0, res = 0;
for(int i = 30; i >= 0; i--){
int s = (x >> i) & 1;
if(son[p][1-s] != 0){
res += (1 << i);
p = son[p][1-s];
}else{
p = son[p][s];
}
}
return res;
}
public static void main(String[] args) throws Exception{
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[] a = new int[n];
for(int i = 0; i < n; i++){
a[i] = sc.nextInt();
insert(a[i]);
}
int res = 0;
for(int num : a){
res = Math.max(res, query(num));
}
System.out.println(res);
}
}