Acwing 143. 最大异或对（Trie 树）

题目链接

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

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

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

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

数据范围
1≤N≤10⁵
0≤Ai<2³¹
输入样例：
3
1 2 3
输出样例：
3

思路：暴力的话两层循环1e10，铁超时，那么按二进制建字典树，优化第二层循环，找出每个数的最大异或值，取max

#include
#include
#include
using namespace std;
const int N = 1e5+10;
int son[N*32][2],idx;
int a[N];
void add(int x)
{
    int p=0;
    for(int i=30;i>=0;i--)
    {
        int u=(x>>i)&1;
        if(!son[p][u]) son[p][u]=++idx;
        p=son[p][u];
    }
}
int query(int x)
{
    int p=0,ans=0;
    for(int i=30;i>=0;i--)
    {
        int u=(x>>i)&1;
        if(son[p][!u]) ans+=1<<i,p=son[p][!u];
        else p=son[p][u];
    }
    return ans;
}
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&a[i]);
        add(a[i]);
    }
    int ans=0;
    for(int i=0;i<n;i++) ans=max(ans,query(a[i]));
    cout<<ans<<endl;
    return 0;
}