421-数组中两个数的最大异或值

题目

421. 数组中两个数的最大异或值 - 力扣（LeetCode） (leetcode-cn.com)

思路

使用二重循环遍历所有可能的组合可以解决该问题，不过时间复杂度过大，会导致运行超时。

为了优化运行速度，对于一个数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;
    }
};