题目地址:
https://www.lintcode.com/problem/maximum-xor-of-two-numbers-in-an-array/description
给定一个非空32位非负整数数组 A A A,求其中两个数异或的最大值。
思路是Trie。将数 x x x转为长32的二进制字符串,插入Trie。假设Trie中已经插入了 A [ 0 : i − 1 ] A[0:i-1] A[0:i−1],当遇到 A [ i ] A[i] A[i]的时候,我们希望寻找 A [ 0 : i − 1 ] A[0:i-1] A[0:i−1]中与 A [ i ] A[i] A[i]异或最大的数,那么可以这样做:也将 A [ i ] A[i] A[i]转为长32的二进制字符串,然后遍历它的同时,从Trie的树根向下走,每次尽量走与 A [ i ] A[i] A[i]那一位不同的一条路(这样能保证最高位尽量是 1 1 1),走到叶子的时候得到的数其实就是 A [ i ] A[i] A[i]与 A [ 0 : i − 1 ] A[0:i-1] 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 ) O(l_A) O(lA)。