LeetCode每日一题 DAY2 剑指 Offer II 067. 最大的异或

        这道题如果使用暴力解法，需要的时间复杂度为 O(n^2)。可以使用更加高效的办法，因为异或的结果是，相同的数位的结果为 0，不同的数位的结果为 1。因此这个问题可以转化为查找问题，可以按照二进制数的数位进行查找。所以需要将数组内的所有数字的数位都保存下来，前缀树就可以完成，只需将每个节点的子节点设置为表示 0 和 1 的两个子节点就行。因为需要返回最大的异或结果，所以应该从最高位开始进行查找。每个数字从最高位开始，若存在与该位不同的数位则选择不同的数位对应的子节点，最终返回该数字对应的最大异或结果。遍历完数组内所有的数字，返回所有异或结果中的最大值就是结果。

完整代码如下，构造前缀树时每个数的长度都为 31 位，所以插入一个数字的时间复杂度为 O(1)，插入所有数字的时间复杂度为 O(n)，同理查找所有数字的最大异或结果的时间复杂度也为 O(n)，故最终整个算法的时间复杂度为 O(n)。

class Trie{
    public:
    Trie* next[2];

    //插入
    void insert(int x){
        Trie* node = this;
        for(int i = 30; i >= 0; i--){
            int u = x >> i & 1;
            if(node -> next[u] == NULL) node -> next[u] = new Trie();
            node = node -> next[u];
        }
    }

    //返回最大异或值
    int maxXOR(int x){
        Trie* node = this;
        int res = 0;

        for (int i = 30; i >= 0; i --){
            int u = x >> i & 1;
            if(node -> next[!u] != NULL)
            {
                res += 1 << i;
                node = node -> next[!u];
            }
            else node = node -> next[u];
        }

        return res;
    }

};

class Solution {
public:
    int findMaximumXOR(vector<int>& nums) {
        Trie* root = new Trie();

        for (int i = 0; i < nums.size(); i ++) root->insert(nums[i]);

        int ret = 0;
        for (int i = 0; i < nums.size(); i ++) ret = max(ret,root->maxXOR(nums[i]));

        return ret;
    }
};

 此题的思想可回溯到 AcWing 143. 最大异或对 该代码也为模板

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

const int N = 100010, M = 3000000;

int a[N], son[M][2], idx;   //a[]存储输入的数 son[][]输入数的二进制的二叉树 idx当前下标

void insert(int x){
    int p = 0;

    for (int i = 30; i >= 0; i --){
        int s = x >> i & 1;                     //将要插入的数x第i位二进制数存入s中
        if (!son[p][s])  son[p][s] = ++ idx;    //如果不存在该节点，创建新节点
        p = son[p][s];                          //指向子节点
    }
}

int query(int x){
    int p = 0, res = 0;     //res用来记录最大异或值

    for (int i = 30; i >= 0; i --){
        int s = x >> i & 1; 
        if (son[p][!s]){    //要想使得异或值最大，第i位二进制数应该相反
            res += 1 << i;  //存在与第i位二进制数相反的数节点（即插入了目的数,每次累加第i位的做大值
            p = son[p][!s];
        }
        else p = son[p][s]; //不存在目的数，则指向自己，并且不累加最大值
    }

    return res;
}

int main(){
    int n;
    cin >> n;

    for (int i = 0; i < n; i ++){
        cin >> a[i];
        insert(a[i]);
    }

    int res = 0;
    for (int i = 0; i < n; i ++ ) res = max(res, query(a[i]));

    printf("%d", res);

    return 0;
}