算法基础之数据结构——Trie最大异或对

题目：
在给定的 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)，这样肯定是过不了的。
我们想到将这些数字分解成二进制表示，然后将这些二进制组成一颗树，因为数据范围是2^31，所以这颗树的深度最大是31，输入的每一个数先变成二进制再插入树中，再用一个查询语句去查询是否有一个数字能与它异或成最大的数字。

查询方式如下：
将查询的树往相反的方向走，例如：如果输入的数字是10100，那么我们希望从根树出发，第一次往0走，然后往1，再往0……。如果这颗树中想走的方向没有节点，那么只好往有方向的节点走，最后找到深度为31的节点，这个节点对应的数就是与它异或最大的数字。

代码如下：

#include<iostream>
#include<algorithm>
using namespace std;

const int N = 100005;
//A存储输入的数,son为组成树的节点，因为深度为31并且输入的数最大为2^31所以为31*N个，b为树的叶节点对应的数
int A[N], son[31*N][2], b[31*N];
//flag为当前数组对应的下标（用数组模拟树），res为最大的异或结果
int flag, res;
//将数组成树
void insert(int x)
{
    int p = 0;//p为当前节点
    for(int i = 30; i >= 0; i --){
        int t = x >> i & 1;//寻找x组成的二进制最左边是0还是1
        //如果节点不存在那么就建立一个节点用于存储，并且数组下标加1
        if(!son[p][t])  son[p][t] = ++ flag;
        //将节点位置更新
        p = son[p][t];
    }
    //循环结束后就找到了树的叶子节点，那么就记录一下这个叶子节点对应的数值
    b[p] = x;
}
//查询树中与当前数异或最大的结果
int query(int x)
{
    int p = 0;//p含义如上
    for(int i = 30; i >= 0; i --){
        int t = x >> i & 1;//作用同上
        //因为想要找最大的异或树，所以将前进的方向取反，如果不存在那么就只好往存在的方向走了
        if(son[p][!t])  p = son[p][!t];
        //存在则最好，继续往前走
        else    p = son[p][t];
    }
    //此时p为当前数值通过树找到与它异或最大的叶子节点处在的位置，b[p]为此叶子节点对应的值
    res = max(res, b[p] ^ x);
}

int main()
{
    int n;
    cin >> n;
    for(int i = 0; i < n; i ++ )    scanf("%d", &A[i]);

    for(int i = 0; i < n; i ++ ){
        insert(A[i]);
        query(A[i]);
    }
    printf("%d",res);
}