题目
思路
使用二重循环遍历所有可能的组合可以解决该问题,不过时间复杂度过大,会导致运行超时。
为了优化运行速度,对于一个数n
, 我们需要尽可能快的找到另一个数m
,使得在所有和n
组合的数中,n^m
的结果最大。
根据异或运算的性质,为了使得计算结果尽可能地大,我们需要保证高比特位有尽可能多地1
,因此,从最高比特位开始(本题目中为第30位(从0开始)),对于n
中为0
的比特位,我们需要尽可能的找该比特位为1
的数,对于n
中为1
的比特位,我们需要尽可能找该比特位为0
的数。
本题目最大数字为2^31 - 1,因此最大数字共计31位
为了实现上述过程的快速查找,我们将数组中所有数字按照最高比特位开始(第30位)的各个比特位的值存储为一棵前缀树。
例如5表示为32位二进制数为
0 ~ 0 1 0 1
(28个0)
将其插入前缀树中为(前缀树结点不表示值,边表示比特0或者1)
T
0/ ---第30位
T
0/ ---第29位
... ...
\1 ---第2位
T
0/ ---第1位
T
\1
T(5) ---第0位
即从i=30开始,第i比特为0则生成一个左子树结点(如果父节点不存在左子树),第i比特为1则生成一个右子树结点(如果父节点不存在右子树)
有了这棵前缀树,对于整数n
,我们就可以通过前缀树来查找m
,查找规则为从第30
比特开始,如果n
的第i
比特为0
,就优先选择右子树(即第i
比特为1
的数),如果没有就选择左子树;如果n
的第i
比特为1
,就优先选择左子树(即第i
比特为0
的数),如果没有则选择右子树。最终得到的数就是同n
异或结果最大的数。
最后我们只需要遍历一次原数组,对于每一个n
按照上述方法求与其异或的最大的m
,再求出所有n^m
的最大值即为最终结果。
代码
struct tier {
tier* right;
tier* left;
tier() {
right = nullptr;
left = nullptr;
}
};
class Solution {
public:
// 向前缀树插入一个元素
void add(tier* root, int num) {
for(int i=30; i>=0; i--) {
int tmp = (num >> i) & 0x1;
if (tmp == 0) {
if (root->left == nullptr) {
root->left = new tier;
}
root = root->left;
} else {
if (root -> right == nullptr) {
root->right = new tier;
}
root = root -> right;
}
}
}
// 在前缀树中查找同num异或结果最大的数
int find_max(tier* root, int num) {
int result = 0;
for(int i=30; i>=0; i--) {
int tmp = (num >> i) & 0x1;
if (tmp == 0) {
if (root->right != nullptr) {
result |= 0x1 << i;
root = root->right;
} else {
root = root->left;
}
} else {
if (root->left != nullptr) {
result |= 0x1 << i;
root = root->left;
} else {
root = root->right;
}
}
}
return result;
}
int findMaximumXOR(vector<int>& nums) {
tier* root = new tier;
int result = 0;
for (int n : nums) {
add(root, n);
}
for (int n : nums) {
result = max(result, find_max(root, n));
}
return result;
}
};