【LeetCode热题100(66/100)】寻找两个正序数组的中位数

题目地址: 链接

思路: 要想O(log(n + m))时间复杂度内实现当前算法。很显然是二分算法,而且需要将两个数组成为一个整体来想。

两个数组都是递增的,并且中位数的左边的内容是一样的,如果将问题转换为获取中位数的下标,那么思路就很清晰了。

  1. 获取两个数组下标。在 nums1 中选取一个下标 mid - 1,由于是两个数组的中位数,而且是两个有序数组,所以另外一个数组的下标是确定的totalN - mid - 1
  2. 选取两个下标的最大值,判断是否大于nums[q1 + 1],如果成立,说明当前成立,可以让左收拢;反之向右收拢。
  3. 重复1,知道 l < r 不成立
  4. 根据 n + m 的奇偶性来返回对应的中位数。
js 复制代码
/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */

var findMedianSortedArrays = function(nums1, nums2) {
    function check_left(mid) {
        if(mid > n) return false;
        if(totalN - mid > m) return true;
        let [q1, q2] = [mid - 1, totalN - mid - 1];
        let maxValue = -Infinity;
        if(q1 >= 0) maxValue = Math.max(nums1[q1], maxValue);
        if(q2 >= 0) maxValue = Math.max(nums2[q2], maxValue);
        if(q1 + 1 < n && maxValue > nums1[q1 + 1]) return true;
        return false;
    }
    let [n, m] = [nums1.length, nums2.length];
    let totalN = n + m + 1 >> 1;
    let isOdd = (n + m) % 2;

    let [l, r] = [0, totalN];
    while(l < r) {
        let mid = l + r >> 1;
        if(check_left(mid)) l = mid + 1;
        else r = mid;
    }

    let ans = -Infinity;
    let [idx1, idx2] = [l - 1, totalN - l - 1]
    if(idx1 >= 0) ans = Math.max(ans, nums1[idx1]);
    if(idx2 >= 0) ans = Math.max(ans, nums2[idx2]);
    if(!isOdd) {
        let otherNum = Infinity
        if(l < n) {
            otherNum = Math.min(otherNum, nums1[l]);
        }
        if(totalN - l < m) {
            otherNum = Math.min(otherNum, nums2[totalN - l]);
        }
        ans = (ans + otherNum) / 2;
    }
    return ans;
};
相关推荐
youngee11几秒前
hot100-48课程表
算法
kesifan5 分钟前
数据结构线性表
数据结构·算法
leo__52014 分钟前
如何计算一个二维地质模型的表面重力值和重力异常
算法
代码游侠25 分钟前
应用——基于Linux的音乐播放器项目
linux·运维·笔记·学习·算法
智驱力人工智能1 小时前
仓库园区无人机烟雾识别:构建立体化、智能化的早期火灾预警体系 无人机烟雾检测 无人机动态烟雾分析AI系统 无人机辅助火灾救援系统
人工智能·opencv·算法·目标检测·架构·无人机·边缘计算
Christo31 小时前
2022-《Deep Clustering: A Comprehensive Survey》
人工智能·算法·机器学习·数据挖掘
天真小巫2 小时前
2025.12.22总结(边工作,边强大)
职场和发展
Yzzz-F2 小时前
牛客周赛round123 G小红出千[补题][滑动窗口]
算法
肆悟先生2 小时前
3.16 含有可变参数的函数
c++·算法