emmm,今天的题虽然明知道不能暴力,但还是使用暴力解法试了一下,虽然已经做了一定的剪枝,但还是超时了,还是好好学习一下字典树吧。新知识–字典树
字典树
具体的讲解可以在知乎等找到。这里不做详细叙述,主要讲一下代码的实现。
解题思路
根据题意,很容易可以得到该条信息——最后进行异或的两个数字中一定包含且只包含一个拥有数组元素中二进制位数最高的元素(数组元素中二进制位数最高的元素不一定唯一)。因为字典树的性质也是从最高位开始,所以使用字典树辅助进行计算。
代码中有几个关键点:
- 数组内元素依次存入字典树;
- 获取每一个数字所对应的最大异或和;
- 获取数组内所有元素中的最大异或和。
第一点使用add函数,根据字典树的性质,如果当前位置为0则是左子树,为1则是右子树
第二点是使用check函数模拟异或过程,因为上面提到
如果当前位置为0则是左子树,为1则是右子树
。而字典树内存储的是当前遍历元素之前的所有元素,模拟异或的过程可以分为以下几种情况:
- 当前元素的当前位置为1:①字典树内当前位置为1(存在右子树),根据异或的性质,所以当前位置的异或结果为0;②字典树内当前位置为0(存在左子树),根据异或的性质,所以当前位置的异或结果为1;
2.当前元素的当前位置为0:①字典树内当前位置为1(存在右子树),根据异或的性质,所以当前位置的异或结果为1;②字典树内当前位置为0(存在左子树),根据异或的性质,所以当前位置的异或结果为0;
这个过程使用x的增加实现。
第三点很容易实现,只是在遍历过程中使用
Math.max
实现。
具体的详见代码内部注释。
代码–来自官方题解的字典树方法
class Solution {
// 字典树的根节点
Trie root = new Trie();
// 最高位的二进制位编号为 30
static final int HIGH_BIT = 30;
public int findMaximumXOR(int[] nums) {
int n = nums.length;
int x = 0;
for (int i = 1; i < n; ++i) {
// 将 nums[i-1] 放入字典树,此时 nums[0 .. i-1] 都在字典树中
add(nums[i - 1]);
// 将 nums[i] 看作 ai,找出最大的 x 更新答案
x = Math.max(x, check(nums[i]));
}
return x;
}
public void add(int num) {
Trie cur = root;//获取根节点
for (int k = HIGH_BIT; k >= 0; --k) {
int bit = (num >> k) & 1;
if (bit == 0) {//从二进制的第31位开始,逐个检验当前输入的二进制形式在当前位置是否为1,为1则存在右子树,为0存在左子树
if (cur.left == null) {//如果没有左子树,则进行创建
cur.left = new Trie();
}
cur = cur.left;
}
else {
if (cur.right == null) {
cur.right = new Trie();
}
cur = cur.right;
}
}
}
public int check(int num) {//模拟异或的过程
Trie cur = root;
int x = 0;
for (int k = HIGH_BIT; k >= 0; --k) {
int bit = (num >> k) & 1;
if (bit == 0) {
// a_i 的第 k 个二进制位为 0,应当往表示 1 的子节点 right 走
if (cur.right != null) {
cur = cur.right;
x = x * 2 + 1;//根据字典树的特性,如果当前位置存在1,因为是从高位开始遍历的,所以先把x*2,然后加1
} else {
cur = cur.left;
x = x * 2;//如果当前位置为0,即只存在左子树,则仅仅把x*2
}
} else {
// a_i 的第 k 个二进制位为 1,应当往表示 0 的子节点 left 走
if (cur.left != null) {
cur = cur.left;
x = x * 2 + 1;
} else {
cur = cur.right;
x = x * 2;
}
}
}
return x;
}
}
class Trie {
// 左子树指向表示 0 的子节点
Trie left = null;
// 右子树指向表示 1 的子节点
Trie right = null;
}