记录个人算法学习4:寻找两个正序数组的中位数

问题描述

给定两个大小分别为 mn 的正序(从小到大)数组 nums1nums2。请你找出并返回这两个正序数组的 中位数

算法的时间复杂度应该为 O(log (m+n))

示例 1:

css 复制代码
输入: nums1 = [1,3], nums2 = [2]
输出: 2.00000
解释: 合并数组 = [1,2,3] ,中位数 2

示例 2:

ini 复制代码
输入: nums1 = [1,2], nums2 = [3,4]
输出: 2.50000
解释: 合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5

⚠️ 注意闭坑

这道题其实是有歧义的一道题,请你找出并返回这两个正序数组的 中位数 ,较真的同学会以为是分别返回这两个数据的中位数,但实际上人家出题肯定不会那么简单,咱们要降为打击的理解下那些考虑不全的"专家",不要较真。

解题思路

  1. O(log (m+n))就应该想到二分法
  2. 两个数组就应该想到双指针
  3. 我们的目的是找到一个分割点(指针),使得 nums1 左侧的元素和 nums2 左侧的元素加起来正好是合并数组的一半
  4. 提取出隐形条件数组1的索引(指针) 加 数组1的索引(指针) 等于 (m+n)/2 这其实是有代数的思想
  5. 可以定义一个变量i作为数组1的索引(指针),每次遍历中用二分法去重新分配i的值

写代码思路

  1. 找出两个数组中,相对短的那一个,这样遍历时效率高一点
  2. 定义一个变量halfLen 用于存储中位数的索引(也等于数组1的索引和数组2的索引的和)
  3. 第一步先暂定中位数的区间在数组1的最小值和最大值之间
js 复制代码
nums1: [1, 3, 5, 7] nums2: [2, 4, 6, 8, 10]
       └──────────┘
  1. 用二分法去判断数组1中间的值偏大还是偏小
js 复制代码
nums1: [1, 3, 5, 7] nums2: [2, 4, 6, 8, 10]
       └──────────┘        └──────────────┘
           └──┘                   └──┘
              ^                      ^

用二分法得到数组1的索引是2,区间的值是[3,5],右侧对应的索引是3,区间是[6,8] 很明显左侧最大值仍然小于右侧的最小值,那么就说明二分法第一次的索引值偏小了,那么中位数就在当前索引值到最后的那个区间内,再次使用二分法数组1的索引设为3(当前索引加最右侧索引再除以2)左侧区间值是[5,7],右侧的区间值是[4,6],

js 复制代码
nums1: [1, 3, 5, 7] nums2: [2, 4, 6, 8, 10]
              └──┘             └──┘
                 ^                ^

这时候刚好数组1左侧最小值小于于数组2右侧的最大值,而且左侧的最大值大于数组2的最小值,说明刚刚好,因为总数组数量是奇数,所以中位数是两个子数组中较大的左端元素

  1. 考虑下特殊情况

代码实现

js 复制代码
    // 确保nums1是较短的数组(对短的数组遍历效率高)
    if (nums1.length > nums2.length) {
        [nums1, nums2] = [nums2, nums1];
    }

    const m = nums1.length;
    const n = nums2.length;
    let imin = 0;
    let imax = m;
    const halfLen = Math.floor((m + n + 1) / 2);
    //用二分法一步步确定中位数所在的区间
    while (imin <= imax) {
        const i = Math.floor((imin + imax) / 2);
        const j = halfLen - i;

        if (i < imax && nums2[j - 1] > nums1[i]) {
            imin = i + 1; // i偏小了,需要增加
        } else if (i > imin && nums1[i - 1] > nums2[j]) {
            imax = i - 1; // i偏大了,需要减小
        } else { // 达到了正确的划分
            let maxLeft = 0;
            if (i === 0) { maxLeft = nums2[j - 1]; }
            else if (j === 0) { maxLeft = nums1[i - 1]; }
            else { maxLeft = Math.max(nums1[i - 1], nums2[j - 1]); }
            if ((m + n) % 2 === 1) { // 如果总长度是奇数,中位数是两个子数组中较大的左端元素
                return maxLeft;
            }

            let minRight = 0;
            if (i === m) { minRight = nums2[j]; }
            else if (j === n) { minRight = nums1[i]; }
            else { minRight = Math.min(nums1[i], nums2[j]); }

            // 如果总长度是偶数,中位数是两个子数组中较大的左端元素和较小的右端元素的平均值
            return (maxLeft + minRight) / 2.0;
        }
    }
相关推荐
知来者逆1 小时前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊1 小时前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北1 小时前
力扣-160.相交链表
算法·leetcode·链表
爱研究的小陈2 小时前
Day 3:数学基础回顾——线性代数与概率论在AI中的核心作用
算法
渭雨轻尘_学习计算机ing2 小时前
二叉树的最大宽度计算
算法·面试
BB_CC_DD3 小时前
四. 以Annoy算法建树的方式聚类清洗图像数据集,一次建树,无限次聚类搜索,提升聚类搜索效率。(附完整代码)
深度学习·算法·聚类
梁下轻语的秋缘4 小时前
每日c/c++题 备战蓝桥杯 ([洛谷 P1226] 快速幂求模题解)
c++·算法·蓝桥杯
CODE_RabbitV4 小时前
【深度强化学习 DRL 快速实践】逆向强化学习算法 (IRL)
算法
mit6.8245 小时前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify5 小时前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法