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;
    }
};
相关推荐
kitesxian2 分钟前
Leetcode448. 找到所有数组中消失的数字(HOT100)+Leetcode139. 单词拆分(HOT100)
数据结构·算法·leetcode
VertexGeek42 分钟前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
石小石Orz43 分钟前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
jiao_mrswang2 小时前
leetcode-18-四数之和
算法·leetcode·职场和发展
qystca2 小时前
洛谷 B3637 最长上升子序列 C语言 记忆化搜索->‘正序‘dp
c语言·开发语言·算法
薯条不要番茄酱2 小时前
数据结构-8.Java. 七大排序算法(中篇)
java·开发语言·数据结构·后端·算法·排序算法·intellij-idea
今天吃饺子2 小时前
2024年SCI一区最新改进优化算法——四参数自适应生长优化器,MATLAB代码免费获取...
开发语言·算法·matlab
是阿建吖!2 小时前
【优选算法】二分查找
c++·算法
王燕龙(大卫)2 小时前
leetcode 数组中第k个最大元素
算法·leetcode
不去幼儿园3 小时前
【MARL】深入理解多智能体近端策略优化(MAPPO)算法与调参
人工智能·python·算法·机器学习·强化学习