421 Maximum XOR of Two Numbers in an Array 数组中两个数的最大异或值

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

给定一个非空数组，数组中元素为 a0, a1, a2, … , an-1，其中 0 ≤ ai < 231 。
找到 ai 和aj 最大的异或 (XOR) 运算结果，其中0 ≤ i, j < n 。
你能在O(n)的时间解决这个问题吗？
示例:
输入: [3, 10, 5, 25, 2, 8]
输出: 28
解释: 最大的结果是 5 ^ 25 = 28.
详见：https://leetcode.com/problems/maximum-xor-of-two-numbers-in-an-array/description/

C++：

class Solution {
public:
int findMaximumXOR(vector<int>& nums)
{
int res = 0, mask = 0;
for (int i = 31; i >= 0; --i)
{
mask |= (1 << i);
unordered_set<int> s;
for (int num : nums)
{
s.insert(num & mask);
}
int t = res | (1 << i);
for (int prefix : s)
{
// 利用了 ^ 的 a ^ b = c，则 b ^ c = a
if (s.count(t ^ prefix))
{
res = t;
break;
}
}
}
return res;
}
};