421. Maximum XOR of Two Numbers in an Array

Given a non-empty array of numbers, a₀, a₁, a₂, … , a_n-1, where 0 ≤ a_i < 2³¹.

Find the maximum result of a_i XOR a_j, where 0 ≤ i, j < n.

Could you do this in O(n) runtime?

Example:

Input: [3, 10, 5, 25, 2, 8]

Output: 28

Explanation: The maximum result is 5 ^ 25 = 28.

思路：

利用异或的特性, 这题可以有更好的解法. 对于异或运算, 我们有

如果a ^ b = c, 那么a ^ c = b.

根据这个定理, 我们从最高位开始找, 我们先将所有数的最高位存到一个Set中. 然后, 我们假设最终答案的最高位为1, 将数列中所有数的最高位和1进行异或运算, 异或得到的结果仍然在Set中, 那么说明最终答案的最高位一定为1, (由定理可得1 ^ x = b <==> b ^ x = 1, 对与数x, 一定存在一个数b, 使得x ^ b = 1), 否则最高位的答案一定为0.

假设我们已经知道最终答案的最高k位为prefix, 我们先将数列中所有数的最高k+1位存入Set中. 然后, 我们假设下一位的值为1, 将数列中所有数的最高k+1位与prefix*2 + 1进行异或运算, 如果异或得到的结果仍然在Set中, 那么说明最终答案的第k+1位一定为1, 否则, 最高位的答案一定为0.

因为x ^ (prefix*2+1) = b　<==> x ^ b = prefix*2+1, 即对于数x, 一定存在一个数b, 使得他们异或的结果为prefix*2+1.

因此, 我们可以对最终答案的32位进行依次判断, 最终得到异或的最大值.

该算法的时间复杂度为O(32n) = O(n).

代码1：

class Solution {
public:
    int findMaximumXOR(vector<int>& nums) {
        int max = 0, mask = 0;
        // test each bit pose, 判断能不能构成所需要的值；
        for (int i = 31; i >= 0; --i) {
            // 每次都在之前的基础上更新mask
            mask |= (1 << i);
            set<int> s;
            for (int num : nums) {
                // add the number which has the mask as its prefix;
                s.insert(num & mask);
            }
            // 假设当前所能达到的最大值是这个temp值；
            int tmp = max | (1 << i);
            for (int prefix : s) {
                if (s.count(tmp ^ prefix)) {
                    max = tmp;
                    break;
                }
            }
        }
        return max;
    }
};

代码2：

class Solution {
public:
    int findMaximumXOR(vector<int>& nums) {
        int res=0,kbit=0;
        for(int i=31;i>=0;i--){
            kbit = kbit | (1<<i);// 准备提起前多少的高位
            set<int> s;
            for(int j:nums) {s.insert(j&kbit);}// 存储高位，所以用&
            int tp=res| (1<<i);//在已获得的高位中再尝试加一位;
            //因为a^b = c -> a^c=b,b^c=a,所以若a^b = c成立，则a^c=b，a从set中来，b亦在set中
            for(int j:s) {
                if(s.count(j^tp)){
                    res=tp;
                    break;
                }
            }
        }
        return res;
    }
};