前缀树求数组两个数最大异或值

421. Maximum XOR of Two Numbers in an Array
     Medium

629

140

Favorite

Share
Given a non-empty array of numbers, a0, a1, a2, … , an-1, where 0 ≤ ai < 231.

Find the maximum result of ai XOR aj, 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树。

根据0和1来建树，最大深度为最大的数的第一位bit到最低位的总位数。

对于[3,10,5,25,2,8]测试用例，如果使用O(n^2)算法，选取3与剩下的数进行异或。这个过程中对于5,10,8的二进制前28位异或结果是一样的，对于10,8而言前30位异或结果是一致的。很容易想到对数组中的数的二进制，构建前缀树。相同前缀的计算结果可复用，就能提升效率。

建树
因为二进制元素只有0,1。考虑使用二叉树来构建前缀树。

根节点为符号位0
从高位到低位依次构建
左子树为1节点，又子树为0节点
[3,10,5,25,2,8] 按照以上规则构建树。

那么这颗树包含了数组中所有的数字，从根节点到叶子节点的一条路径表示数组中的一个十进制数的二进制编码。

找最大异或值
对于25 (0000 0000 0000 0000 0000 0000 0001 1001) 而言，从第31-6位（都是正数，不考虑符号位），都为0,，且数组中的数字31-6位都为0，因此最大异或结果31-6位只能为0。
当前使得异或结果最大的数x为0000 0000 0000 0000 0000 0000 000*****
当前指针curNode指向第图中根节点。
从第5位开始：
5 1 x的第5位为0，则结果最大。应选择右分支，curNode = curNode->right
4 1 x的第4位为0，则结果最大。应选择右分支，curNode = curNode->right
3 0 x的第3位为1，则结果最大。应选择左分支，curNode = curNode->left
2 0 x的第2位为1，则结果最大。应选择左分支，但树中左分支为null，只能选择右分支curNode = curNode->right 于是x的第2位为0。
1 1 x的第1位为0，则结果最大。应选择右分支，但树中右分支为null，只能选择左分支curNode = curNode->left 于是x的第1位为1。
找到的x为5（00101）。

代码

class Solution {
   
    TreeNode root = new TreeNode(0);

    public int findMaximumXOR(int[] nums) {
        int max = findMax(nums);
        int maxBit = 32 - Integer.numberOfLeadingZeros(max);
        //build prefix tree
        TreeNode currNode = root;
        for (int i = 0; i < nums.length; i++) {
            for (int j = maxBit -1; j >=0 ; j--) {
                int tmp = (nums[i] >>j)& 1;
                if(tmp == 0)
                {
                    if(currNode.left == null)
                    {
                        currNode.left = new TreeNode(0);
                    }
                    currNode = currNode.left;
                }else {
                    if(currNode.right == null)
                    {
                        currNode.right = new TreeNode(1);
                    }
                    currNode = currNode.right;
                }
            }
            // finished for nums[i], will do it for nums[i++];
            currNode = root;
        }

        //find max xor value
        int maxXOR = 0;

        for (int i = 0; i < nums.length; i++) {
            int res = 0;
            for (int j = maxBit -1; j >=0; j--) {
                int tmp = (nums[i] >>j)& 1;
                if(currNode.left != null && currNode.right != null)
                {
                    if(tmp == 0)
                    {
                        currNode = currNode.right;
                    }else {
                        currNode = currNode.left;
                    }
                }else {
                    currNode = currNode.left == null ? currNode.right : currNode.left;
                }

                res += (tmp^currNode.val) << j;
            }
            currNode = root;
            maxXOR = maxXOR > res ? maxXOR : res;
        }
        return maxXOR;
    }
    private int findMax(int[] nums)
    {
        int max = nums[0];
        for (int i = 1; i < nums.length; i++) {
            if(nums[i] > max)
            {
                max = nums[i];
            }
        }
        return max;
    }
    private static class TreeNode{
        int val;
        TreeNode left,right;
        TreeNode(int val)
        {
            this.val = val;
        }
    }
}