[LeetCode]421. 数组中两个数的最大异或值

题目

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

421. 数组中两个数的最大异或值
给你一个整数数组 nums ，返回 nums[i] XOR nums[j] 的最大运算结果，其中 0 ≤ i ≤ j < n 。

进阶：你可以在 O(n) 的时间解决这个问题吗？

 

示例 1：

输入：nums = [3,10,5,25,2,8]
输出：28
解释：最大运算结果是 5 XOR 25 = 28.
示例 2：

输入：nums = [0]
输出：0
示例 3：

输入：nums = [2,4]
输出：6
示例 4：

输入：nums = [8,10,2]
输出：10
示例 5：

输入：nums = [14,70,53,83,49,91,36,80,92,51,66,70]
输出：127
 

提示：

1 <= nums.length <= 2 * 104
0 <= nums[i] <= 231 - 1

解法

public int findMaximumXOR(int[] nums) {
    for (int x : nums) insert(x);
    int maxx = 0;
    for (int x : nums) {
        maxx = Math.max(maxx, query(x));
    }
    return maxx;
}


Trie root = new Trie();

public void insert(int x) {
    Trie cur = root;
    for (int i = 30; i >= 0; i--) {
        int u = x >> i & 1;//找打二进制的第i位上判断是0还是1
        if (cur.children[u] == null) {
            cur.children[u] = new Trie();
        }
        cur = cur.children[u];
    }
}


public int query(int x) {
    Trie cur = root;
    int res = 0;
    for (int i = 30; i >= 0; i--) {
        int u = x >> i & 1;//找打二进制的第i位上判断是0还是1
        if (cur.children[u ^ 1] != null) {//如果当前位u的另外一个分支可以走的，那就走这个分支
            res += (1 << i);//相当于当前的值左移i位叠加到res上
            cur = cur.children[u ^ 1];
        } else {//另外一个分支是空，只能和当前的分支一起前行
            cur = cur.children[u];
        }
    }
    return res;
}


class Trie {
    Trie[] children = new Trie[2];
}