45、【树和二叉树】Trie（前缀树/字典树）最大异或对（C/C++版）

前置知识

44、【树和二叉树】Trie（前缀树/搜索树）字符串统计

题目描述

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

输入格式

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

输出格式

输出一个整数表示答案。

数据范围

1≤N≤10⁵,
0≤Ai<2³¹

输入样例：

3
1 2 3

输出样例：

3

题目分析

首先分析异或运算，当两个数所在位数的值相同时，将会取0，相异时将会取1。

因此若想让数值最大，第一要让位数最多，第二要让相同位数下高位的1最多。

因此比较时，从选取数进行比较时，从最高位开始比较更方便。首先，将数由最高位至最低位的顺序依次存取到Trie树中，由于规定了数的位数范围因此，都将其存储到32个空间中，高位不足的补0填充。

选取规则： 选取一个数，然后再选取另一个数从最高位开始比较，每次的最优选择是另一个被选取的数要与选取的数的最高位的数值相反。若不存在这样的数，则选取数值位相同的数，然后再从下一位比较，用与之前相同的规则，选取另一个数。

参考文章：AcWing 143. 最大异或对

算法实现

#include <iostream>
//#include <algorithm>

//using namespace std;

const int N = 1e5 +10, M = 31 * N;
int n;
int a[N];
int son[M][2], idx;

void insert(int x){
    int p = 0;
    for(int i = 30; ~i; i--){   // ~i 等价于 i >= 0，因为i= -1时，取反为00..0
        int u = x >> i & 1;     // 将x右移i位，可获取第i位数值为1还是为0
        if(!son[p][u])      son[p][u] = ++idx;
        p = son[p][u];
    }
}

int query(int x){
    int p = 0, res = 0;
    for(int i = 30; ~i; i--){
        int u = x >> i & 1;
        if(son[p][!u]){     // 先寻找相异的位
            p = son[p][!u];
            res = res * 2 + !u;     // 左移一位，最后一位为!u
        }else{              // 若未找到相异的位，则从同位下寻找
            p = son[p][u];
            res = res * 2 + u;
        }
    }
    return res ^ x;     // 返回异或后的情况
}

int main(){
    int n;      scanf("%d", &n);
    for(int i = 0; i < n; i++){
        scanf("%d", &a[i]);
        insert(a[i]);
    }      

    int res = 0;
    for(int i = 0; i < n; i++){
        //res = std::max(res, query(a[i]));
        //if(res < query(a[i]))     res = query(a[i]);
        res < query(a[i]) ? res = query(a[i]) : res = res;
    }

    printf("%d", res);

    return 0;
}