【Lintcode】1248. Maximum XOR of Two Numbers in an Array

题目地址：

https://www.lintcode.com/problem/maximum-xor-of-two-numbers-in-an-array/description

给定一个非空32位非负整数数组$A$，求其中两个数异或的最大值。

思路是Trie。将数$x$转为长32的二进制字符串，插入Trie。假设Trie中已经插入了$A[0:i-1]$，当遇到$A[i]$的时候，我们希望寻找$A[0:i-1]$中与$A[i]$异或最大的数，那么可以这样做：也将$A[i]$转为长32的二进制字符串，然后遍历它的同时，从Trie的树根向下走，每次尽量走与$A[i]$那一位不同的一条路（这样能保证最高位尽量是$1$），走到叶子的时候得到的数其实就是$A[i]$与$A[0:i-1]$中的数的最大异或值。代码如下：

public class Solution {
    
    class Node {
        Node[] next;
    
        public Node() {
            next = new Node[2];
        }
    }
    
    private Node root;
    
    /**
     * @param nums:
     * @return: the maximum result of ai XOR aj, where 0 ≤ i, j < n
     */
    public int findMaximumXOR(int[] nums) {
        // Write your code here
        root = new Node();
        int res = 0;
        
        // 先把nums[0]插入进Trie
        insert(getBin(nums[0]));
        
        for (int i = 1; i < nums.length; i++) {
            String bin = getBin(nums[i]);
            res = Math.max(res, findMaxXor(bin));
            insert(bin);
        }
        
        return res;
    }
    
    private int findMaxXor(String s) {
        Node cur = root;
        int res = 0;
    
        for (int i = 0; i < s.length(); i++) {
            int val = s.charAt(i) - '0';
            res <<= 1;
            
            // 如果存在val ^ 1这条路，就走，否则走val这条路
            if (cur.next[val ^ 1] != null) {
                res++;
                cur = cur.next[val ^ 1];
            } else {
                cur = cur.next[val];
            }
        }
        
        return res;
    }
    
    private void insert(String s) {
        Node cur = root;
    
        for (int i = 0; i < s.length(); i++) {
            int dig = s.charAt(i) - '0';
            if (cur.next[dig] == null) {
                cur.next[dig] = new Node();
            }
            
            cur = cur.next[dig];
        }
    }
    
    private String getBin(int x) {
        String s = Integer.toBinaryString(x);
        StringBuilder sb = new StringBuilder();
        // 不足32位要补齐
        for (int i = s.length(); i < 31; i++) {
            sb.append('0');
        }
        
        return sb.append(s).toString();
    }
}

时空复杂度$O(l_A)$。