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;
    }
};
相关推荐
有梦想的骇客2 小时前
书籍将正方形矩阵顺时针转动90°(8)0605
线性代数·算法·矩阵
有梦想的骇客2 小时前
书籍“之“字形打印矩阵(8)0609
java·算法·矩阵
Chenyu_3103 小时前
12.找到字符串中所有字母异位词
c语言·数据结构·算法·哈希算法
苏三福3 小时前
yolo11-seg ultralytics 部署版本
算法·yolo11
wuqingshun3141596 小时前
蓝桥杯 冶炼金属
算法·职场和发展·蓝桥杯
jndingxin7 小时前
OpenCV CUDA模块光流计算-----实现Farneback光流算法的类cv::cuda::FarnebackOpticalFlow
人工智能·opencv·算法
编程绿豆侠7 小时前
力扣HOT100之栈:394. 字符串解码
java·算法·leetcode
朝朝又沐沐7 小时前
基于算法竞赛的c++编程(18)string类细节问题
开发语言·c++·算法
记得早睡~8 小时前
leetcode73-矩阵置零
数据结构·leetcode·矩阵