421. 数组中两个数的最大异或值/字典树【leetcode】

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

给你一个整数数组 nums ,返回 nums[i] XOR nums[j] 的最大运算结果,其中 0 ≤ i ≤ j < n 。

示例 1:

输入:nums = [3,10,5,25,2,8]

输出:28

解释:最大运算结果是 5 XOR 25 = 28.

示例 2:

输入:nums = [14,70,53,83,49,91,36,80,92,51,66,70]

输出:127

提示:

1 <= nums.length <= 2 * 105

0 <= nums[i] <= 2的31次方 - 1

字典树

思路:可以将每个数变成最高31位的二进制数,建立字典树,为使异或值最大,应该尽可能找不一样的,如果该位为0先找1,该位为1先找0

cpp 复制代码
class Solution {
public:
    //记录某个id下下个数为0或者1的id
    int a[200005*31][2];
    
    //记录某个id代表的十进制数 是唯一的
    int cnt[200005*31];
    
    //按顺序赋予id,代表树的一个节点
    int id=0;
    
    int findMaximumXOR(vector<int>& nums) {
        for(int i=0;i<nums.size();i++){
            insert(nums[i]);
        }
        int res=0;
        for(int i=0;i<nums.size();i++){
            res = max(res,find(nums[i]));
        }
        return res;
    }

    //建树
    void insert(int b){
        int p=0;
        for(int i=30;i>=0;i--){
            int x=(b>>i)&1;//获得第i位的值
            if(a[p][x]==0) a[p][x]=++id;//新建节点
            p=a[p][x];
        }
        cnt[p]=b;//记录该节点的十进制数
    }

    //找树中和b异或的最大值
    int find(int b){
        int p=0;
        int big=0;
        for(int i=30;i>=0;i--){
            int x=(b>>i)&1;
            if(x==1){
                if(a[p][0]!=0) p=a[p][0];
                else if(a[p][1]!=0) p=a[p][1];
            }
            if(x==0){
                if(a[p][1]!=0) p=a[p][1];
                else if(a[p][0]!=0) p=a[p][0];
            }
        }
        big=max(big,b^cnt[p]);
        return big;
    }
};
相关推荐
YuTaoShao7 分钟前
【LeetCode 热题 100】994. 腐烂的橘子——BFS
java·linux·算法·leetcode·宽度优先
Wendy14418 小时前
【线性回归(最小二乘法MSE)】——机器学习
算法·机器学习·线性回归
拾光拾趣录8 小时前
括号生成算法
前端·算法
渣呵9 小时前
求不重叠区间总和最大值
算法
拾光拾趣录9 小时前
链表合并:双指针与递归
前端·javascript·算法
好易学·数据结构9 小时前
可视化图解算法56:岛屿数量
数据结构·算法·leetcode·力扣·回溯·牛客网
香蕉可乐荷包蛋10 小时前
AI算法之图像识别与分类
人工智能·学习·算法
chuxinweihui10 小时前
stack,queue,priority_queue的模拟实现及常用接口
算法
tomato0910 小时前
河南萌新联赛2025第(一)场:河南工业大学(补题)
c++·算法
墨染点香11 小时前
LeetCode Hot100【5. 最长回文子串】
算法·leetcode·职场和发展