Leetcode 421 数组中两个数的最大异或值

题目链接
题目描述
给你一个整数数组 $nums$，返回 $nums[i]$ $XOR$ $nums[j]$ 的最大运算结果，其中 $0\le i\le j<n$ 。

示例 1：

输入：nums = [3,10,5,25,2,8]
输出：28
解释：最大运算结果是 5 XOR 25 = 28.

提示：
$1\le nums.length\le 2\ast 10^4$
$0\le nums[i]\le 231-1$

$Trie$树
在 1~$x$ 中找个数 $num$ 与 $x$ 异或值最大。

class Solution {
public:
    vector<vector<int>> son;
    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] = son.size();
                son.push_back({0, 0});
            }
            p = son[p][u];
        }
    }
    int query(int x)
    {
        int p = 0, res = 0;
        for(int i = 30; i >= 0; --i)
        {
            int u = x >> i & 1;
            if(son[p][!u])
            {
                p = son[p][!u];
                res = (res << 1) + !u;
            }
            else
            {
                p = son[p][u];
                res = (res << 1) + u;
            }
        }
        return res^x;
    }
    int findMaximumXOR(vector<int>& nums) {
        son.push_back({0, 0});
        int res = 0;
        for(auto x : nums){
            res = max(res, query(x));
            add(x);
        }
        return res;
    }
};