最大异或对（巧用Trie树的数据结构优化算法）

马走太阳

于 2022-10-04 19:50:11 发布

阅读量276

点赞数

文章标签：算法 c++

本文链接：https://blog.csdn.net/WCFNB/article/details/127165234

版权

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

输入格式
第一行输入一个整数 N。

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

输出格式
输出一个整数表示答案。

数据范围
1≤N≤10^5,
0≤Ai<2^31
输入样例：
3
1 2 3
输出样例：
3

想法：暴力做时间复杂度是O(n^2)会TLE，就要对算法考虑优化问题，将数看成2进制01串，从高位往低位开始去建Trie树，然后根据数Ai去找与其异或最大的数，往与当前位不同的方向去找，没有与之相异的位的话就走与之相同的位方向，一直找到最后。

#include<iostream>
using namespace std;
const int N=1e5+10,M=3e6;
int son[M][2],idx;
void insert(int x){
    int p=0;
   for(int i=30;~i;--i){
       if(!son[p][x>>i&1])son[p][x>>i&1]=++idx;
       p=son[p][x>>i&1];
   }
}
int XOR(int a){
    int res=0;
    int p=0;
   for(int i=30;~i;--i){
       if(son[p][!(a>>i&1)]){
           res+=1<<i;
           p=son[p][!(a>>i&1)];
       }
       else p=son[p][(a>>i&1)];
   }
   return res;
}
int main(){
    int n;
    cin>>n;
    int a[N];
    int res=0;
    for(int i=1;i<=n;++i){cin>>a[i];insert(a[i]);}
    for(int i=1;i<=n;++i){
        res=max(res,XOR(a[i]));
    }
    cout<<res<<endl;
    return 0;
}

要知道每个<2^31数最多有31位,而有n个数，故最多有310000个结点，故int Son[310000][2];只有尽可能都不同才能是所获得的数最大。