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

emmm,今天的题虽然明知道不能暴力,但还是使用暴力解法试了一下,虽然已经做了一定的剪枝,但还是超时了,还是好好学习一下字典树吧。新知识–字典树
在这里插入图片描述

字典树

具体的讲解可以在知乎等找到。这里不做详细叙述,主要讲一下代码的实现。
在这里插入图片描述

解题思路

根据题意,很容易可以得到该条信息——最后进行异或的两个数字中一定包含且只包含一个拥有数组元素中二进制位数最高的元素(数组元素中二进制位数最高的元素不一定唯一)。因为字典树的性质也是从最高位开始,所以使用字典树辅助进行计算。
代码中有几个关键点:

  1. 数组内元素依次存入字典树;
  2. 获取每一个数字所对应的最大异或和;
  3. 获取数组内所有元素中的最大异或和。

第一点使用add函数,根据字典树的性质,如果当前位置为0则是左子树,为1则是右子树

第二点是使用check函数模拟异或过程,因为上面提到如果当前位置为0则是左子树,为1则是右子树。而字典树内存储的是当前遍历元素之前的所有元素,模拟异或的过程可以分为以下几种情况:

  1. 当前元素的当前位置为1:①字典树内当前位置为1(存在右子树),根据异或的性质,所以当前位置的异或结果为0;②字典树内当前位置为0(存在左子树),根据异或的性质,所以当前位置的异或结果为1;
    2.当前元素的当前位置为0:①字典树内当前位置为1(存在右子树),根据异或的性质,所以当前位置的异或结果为1;②字典树内当前位置为0(存在左子树),根据异或的性质,所以当前位置的异或结果为0;
    这个过程使用x的增加实现。

第三点很容易实现,只是在遍历过程中使用Math.max实现。

具体的详见代码内部注释。

代码–来自官方题解的字典树方法

class Solution {
     // 字典树的根节点
    Trie root = new Trie();
    // 最高位的二进制位编号为 30
    static final int HIGH_BIT = 30;

    public int findMaximumXOR(int[] nums) {
        int n = nums.length;
        int x = 0;
        for (int i = 1; i < n; ++i) {
            // 将 nums[i-1] 放入字典树,此时 nums[0 .. i-1] 都在字典树中
            add(nums[i - 1]);
            // 将 nums[i] 看作 ai,找出最大的 x 更新答案
            x = Math.max(x, check(nums[i]));
        }
        return x;
    }

    public void add(int num) {
        Trie cur = root;//获取根节点
        for (int k = HIGH_BIT; k >= 0; --k) {
            int bit = (num >> k) & 1;
            if (bit == 0) {//从二进制的第31位开始,逐个检验当前输入的二进制形式在当前位置是否为1,为1则存在右子树,为0存在左子树
                if (cur.left == null) {//如果没有左子树,则进行创建
                    cur.left = new Trie();
                }
                cur = cur.left;
            }
            else {
                if (cur.right == null) {
                    cur.right = new Trie();
                }
                cur = cur.right;
            }
        }
    }

    public int check(int num) {//模拟异或的过程
        Trie cur = root;
        int x = 0;
        for (int k = HIGH_BIT; k >= 0; --k) {
            int bit = (num >> k) & 1;
            if (bit == 0) {
                // a_i 的第 k 个二进制位为 0,应当往表示 1 的子节点 right 走
                if (cur.right != null) {
                    cur = cur.right;
                    x = x * 2 + 1;//根据字典树的特性,如果当前位置存在1,因为是从高位开始遍历的,所以先把x*2,然后加1
                } else {
                    cur = cur.left;
                    x = x * 2;//如果当前位置为0,即只存在左子树,则仅仅把x*2
                }
            } else {
                // a_i 的第 k 个二进制位为 1,应当往表示 0 的子节点 left 走
                if (cur.left != null) {
                    cur = cur.left;
                    x = x * 2 + 1;
                } else {
                    cur = cur.right;
                    x = x * 2;
                }
            }
        }
        return x;
    }
}

class Trie {
    // 左子树指向表示 0 的子节点
    Trie left = null;
    // 右子树指向表示 1 的子节点
    Trie right = null;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值