算法题解—Trie树—最大异或对

最大异或对

在给定的N个整数A1，A2……AN中选出两个进行xor（异或）运算，得到的结果最大是多少？

输入格式

第一行输入一个整数N。

第二行输入N个整数A1～AN。

输出格式

输出一个整数表示答案。

数据范围
1≤N≤105,
0≤Ai<231

输入样例：

3
1 2 3

输出样例：

3

题解
该题是一个字典树/Trie树的典型题目
首先容易想到暴力的做法

for(int i = 0;i < n;i ++){
	for(int j = 0;j < i;j ++){
		res = max(res,a[i]^a[j]);
	}
}

时间复杂度O(n^2)，一定会超时。所以我们考虑如何优化

一个整数可以用32位二进制来表示，题目中数的范围是2的31次方，那么最大就有31位
根据异或运算（相同为0，不同为1），如果两个数高位的二进制是不同的，那么异或结果比低位不同要大
例如：A:101(5)
那么我们只需要依次根据高位二进制不同来筛选集合中的数，例如：B:001 C:111 D:010
第一次筛掉C，剩下B,D。第二次筛掉B，剩下D，那么D就是与其异或得到的值最大
如果集合中所有数的某一位数都一样，即无对称分支，那么我们不需要筛选，继续到下一位

所以可以总结出这本质上是Trie树的问题，每一个结点就两个分支，0和1，我们每一次只需要找与当前位相反的分支递归下去，如果没有相反的分支，就单分支递归下去，最后到叶节点，得到的数一定与基准数异或一定最大

代码

#include <iostream>
using namespace std;

int n;
const int N = 100010, M = 31 * 100010;
int a[N];
int son[M][2],idx;

void insert(int x){
    
    int p = 0;
    for(int i = 30;i >= 0;i --){
        int t = x >> i & 1;
        if(!son[p][t]) son[p][t] = ++ idx;
        p = son[p][t];
    }
}

int query(int x){
    int res = 0;
    int p = 0;
    for(int i = 30;i >= 0;i --){
        int t = x >> i & 1;
        if(son[p][!t]){
            p = son[p][!t];
            res = res * 2 + !t;
        }
        else {
            p = son[p][t];
            res = res * 2 + t;
        }
    }
    return res;
}

int main()
{
    scanf("%d",&n);
    
    for(int i = 0;i < n;i ++) scanf("%d",&a[i]);
    
    int res = 0;
    for(int i = 0;i < n;i ++){
        insert(a[i]);
        res = max(res,query(a[i]) ^ a[i]);
    }
    
    printf("%d\n",res);
    
    return 0;
}