LC 数组中两个数的最大异或值

题目

代码

// 参考这个链接：https://leetcode-cn.com/problems/maximum-xor-of-two-numbers-in-an-array/solution/qian-zhui-shu-by-kai1314/
// https://leetcode-cn.com/problems/maximum-xor-of-two-numbers-in-an-array/solution/li-yong-yi-huo-yun-suan-de-xing-zhi-tan-xin-suan-f/
// 根据图进行理解
// 题目中说到最大数小于2的31次方，所以for循环31次即可
// 先确定高位，再确定低位（贪心算法），才能保证这道题的性质
// 1位接着1位去确定这个数的大小
// 主要思路：构造具备公共前缀的字典树，每次遍历数字，如果当前数字高位为1时，判断是否存在0的支路
// 如果当前数字低位数字为0时，判断当前是否存在1的支路。
// 依次遍历完所有字符，最终得到最大的result
class Solution {
    struct Node {
        Node *one{nullptr};
        Node *zero{nullptr};
    };
    void BuildTrie(Node *root, int num) {
        for (int i = 31; i >= 0; i--) {
            int bitVal = (num >> i) & 1;
            if (bitVal == 1) {
                if (root->one == nullptr) {
                    root->one = new Node();
                }
                root = root->one;
            } else {
                if (root->zero == nullptr) {
                    root->zero = new Node();
                }
                root = root->zero;
            }
        }
    }
public:
    int findMaximumXOR(vector<int>& nums) {
        Node *root = new Node();
        for (auto num : nums) {
            BuildTrie(root, num);
        }

        int res = 0;
        for (auto num : nums) {
            int tmpVal = 0;
            Node *pTmp = root;
            for (int i = 31; i >= 0; i--) {
                int bitVal = (num >> i) & 1;
                if (bitVal == 1) {
                    if (pTmp->zero != nullptr) {
                        pTmp = pTmp->zero;
                        tmpVal += (1 << i);
                    } else {
                        pTmp = pTmp->one;
                    }
                } else {
                    if (pTmp->one != nullptr) {
                        pTmp = pTmp->one;
                        tmpVal += (1 << i);
                    } else {
                        pTmp = pTmp->zero;
                    }
                }
                
            }
            res = max(res, tmpVal);
        }

        return res;
    }
};