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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值