寻找两个有序数组的中位数

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

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

示例 1:

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

示例 2:

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

在编程中,我们常遇到需要找两个已排序数组的中位数的问题。这个问题看似简单,但当数组非常大时,直接合并数组再计算中位数的时间复杂度可能会很高。今天,我们将介绍一个高效的算法,利用二分查找技术来降低时间复杂度。

问题背景

给定两个已排序的数组 nums1nums2,我们的目标是找出这两个数组合并后的中位数。可以通过以下几种方法来解决,但最有效的方法是使用二分查找。

算法思想
  1. 保证较短的数组参与二分查找

    首先,我们保证对较短的数组进行二分查找,这样可以减少查找的范围。我们交换 nums1nums2,确保 nums1 总是较短的那个数组。

  2. 虚拟边界的引入

    为了方便判断边界条件,我们在两个数组的两端引入虚拟值:Integer.MIN_VALUEInteger.MAX_VALUE。这些虚拟值的引入简化了对数组两端的边界检查,避免了处理边界时的复杂逻辑。

  3. 二分查找

    通过对较短的数组 nums1 进行二分查找,寻找合适的切分点。每次切分后,检查切分是否满足中位数的条件:左边部分的最大值不大于右边部分的最小值。

  4. 计算中位数

    一旦找到了合适的切分点,就可以根据数组的总长度来计算中位数。如果总长度是奇数,返回左边部分的最大值;如果是偶数,返回左边最大值和右边最小值的平均值。

算法实现
java 复制代码
class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if (nums1.length > nums2.length) {
            int[] tmp = nums1;
            nums1 = nums2;
            nums2 = tmp;
        }
        int m = nums1.length;
        int n = nums2.length;
        int[] a = new int[m + 2];
        int[] b = new int[n + 2];
        a[0] = b[0] = Integer.MIN_VALUE;
        a[m + 1] = b[n + 1] = Integer.MAX_VALUE;
        System.arraycopy(nums1, 0, a, 1, m);
        System.arraycopy(nums2, 0, b, 1, n);

        int i = 0;
        int j = (m + n + 1) / 2;
        while (true) {
            if (a[i] <= b[j + 1] && a[i + 1] > b[j]) {
                int max1 = Math.max(a[i], b[j]);
                int min2 = Math.min(a[i + 1], b[j + 1]);
                return (m + n) % 2 > 0 ? max1 : (max1 + min2) / 2.0;
            }
            i++;
            j--;
        }
    }
}
结论

这段代码通过二分查找高效地找出了中位数,时间复杂度为 O(log(min(n, m))),避免了合并数组的高复杂度操作。这种方法是求解两个已排序数组中位数的最佳方案。

希望这篇文章对你理解和应用该算法有所帮助!

相关推荐
满分观察网友z6 分钟前
告别烦人的“三连发”:我的智能评论系统过滤之旅(1957. 删除字符使字符串变好)
算法
小新学习屋7 分钟前
《剑指offer》-数据结构篇-哈希表/数组/矩阵/字符串
数据结构·leetcode·哈希表
满分观察网友z9 分钟前
滑动窗口下的极限挑战:我在实时数据流中挖掘最大价值分(1695. 删除子数组的最大得分)
算法
山烛26 分钟前
KNN 算法中的各种距离:从原理到应用
人工智能·python·算法·机器学习·knn·k近邻算法·距离公式
guozhetao39 分钟前
【ST表、倍增】P7167 [eJOI 2020] Fountain (Day1)
java·c++·python·算法·leetcode·深度优先·图论
吃着火锅x唱着歌41 分钟前
LeetCode 611.有效三角形的个数
算法·leetcode·职场和发展
CHANG_THE_WORLD4 小时前
金字塔降低采样
算法·金字塔采样
不知天地为何吴女士6 小时前
Day32| 509. 斐波那契数、70. 爬楼梯、746. 使用最小花费爬楼梯
算法
小坏坏的大世界6 小时前
C++ STL常用容器总结(vector, deque, list, map, set)
c++·算法
励志要当大牛的小白菜8 小时前
ART配对软件使用
开发语言·c++·qt·算法