【每日一题Day368】LC421数组中两个数的最大异或值 | 字典树

最新推荐文章于 2024-09-03 20:45:39 发布

TIkitianya

最新推荐文章于 2024-09-03 20:45:39 发布

阅读量276

点赞数

分类专栏：每日一题字典树文章标签：算法 leetcode

本文链接：https://blog.csdn.net/Tikitian/article/details/134218772

版权

每日一题同时被 2 个专栏收录

371 篇文章 1 订阅

订阅专栏

字典树

5 篇文章 0 订阅

订阅专栏

数组中两个数的最大异或值【LC421】

Given an integer array nums, return the maximum result of nums[i] XOR nums[j], where 0 <= i <= j < n.

字典树

思路
- 首先使用字典树存储数组中元素的二进制形式，由于 $\le 2^{31}-1$ ，因此需要用32位二进制表示；
- 然后枚举 $n u m s [i]$ ，并将 $n u m s [0, i - 1]$ 作为 $n u m s [j]$ 放入字典树中，找到使异或值最大的 $n u m s [j]$ 返回，然后更新结果
  - 如果将 $n u m s [0, i]$ 放入字典树，那么当 $i = 0$ 时，只能返回自身，异或最大值为0
- 如何找到使异或值最大的 $n u m s [j]$ ？
  
  从字典树的根节点开始遍历，并从最高位往低位查询，优先让每一位的异或结果为1，即优先匹配与之不同的二进制位【局部最优】，这样才能使最终的异或结果最大【全局最优】，

实现

class Solution {
    class TrieNode{
        TrieNode[] next = new TrieNode[2];
    }
    TrieNode root = new TrieNode();
    public int findMaximumXOR(int[] nums) {
        TrieNode root;
        int res = 0;
        for (int num : nums){
            add(num);
            res = Math.max(search(num) ^ num, res);
            
        }
        return res;
    }
    public void add(int x){
        TrieNode p = root;
        for (int i = 31; i >= 0; i--){
            int u = (x >> i) & 1;
            if (p.next[u] == null) p.next[u] = new TrieNode();
            p = p.next[u];
        }
    } 
    public int search(int x){
        int res = 0;
        TrieNode p = root;
        for (int i = 31; i >= 0; i--){
            int u = (x >> i) & 1, v = 1 - u; 
            if (p.next[v] != null) {
                res |= (v << i);
                p = p.next[v];
            }else {
                res |= (u << i);
                p = p.next[u];
            }         
        }
        return res;
    }
}