421. Maximum XOR of Two Numbers in an Array

最新推荐文章于 2021-08-02 01:33:06 发布

一头食量超大的小猪

最新推荐文章于 2021-08-02 01:33:06 发布

阅读量380

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/sinat_17451213/article/details/53454838

版权

leetcode 专栏收录该内容

118 篇文章 0 订阅

订阅专栏

Given a non-empty array of numbers, a₀, a₁, a₂, … , a_n-1, where 0 ≤ a_i < 2³¹.

Find the maximum result of a_i XOR a_j, where 0 ≤ i, j < n.

Could you do this in O(n) runtime?

Example:

Input: [3, 10, 5, 25, 2, 8]

Output: 28

Explanation: The maximum result is 5 ^ 25 = 28.

分析：采用trie，字典树，即将这些数根据其bit位依次从最高位开始存储。root不存储任何元素。

一边存储的时候一边计算其与已经存储在字典树中的值得最大值。

public class Solution {
    class Trie{
        Trie[] child=new Trie[2];
    }
    public int findMaximumXOR(int[] nums) {
        int n=nums.length;
        
        //construct trie tree
        Trie root=new Trie();
        int temp;
        int max=Integer.MIN_VALUE;
        for(int num:nums){
            temp=0;
            Trie current=root;
            Trie num_trie=root;
            for(int i=31;i>=0;i--){
                int flag=(num>>>i)&1;
                if(current.child[flag]==null)
                    current.child[flag]=new Trie();
                current=current.child[flag];
                
                if(num_trie.child[1^flag]!=null){
                    temp+=(1<<i);
                    num_trie=num_trie.child[1^flag];
                }else{
                    num_trie=num_trie.child[flag];
                }
            }
            max=max>temp?max:temp;
        }
        return max;
    }
}