4. Median of Two Sorted Arrays

最新推荐文章于 2026-04-18 11:44:27 发布

原创最新推荐文章于 2026-04-18 11:44:27 发布 · 247 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#算法 #二分法

leetcode-java 专栏收录该内容

146 篇文章

订阅专栏

本文介绍了一种使用二分查找算法解决合并两个有序数组并找到中位数的问题的方法。通过合理切分两个数组，确保左侧元素数量等于或比右侧多一，且左侧最大值不大于右侧最小值，从而在O(log(m+n))的时间复杂度内找到中位数。

1题目理解

输入：2个已经排序号的int数组nums1,nums2
输出：这两个数组合并后的中位数
要求：m是nums1的长度，n是nums2的长度。时间复杂度应该是O(log(m+n))。
例子:
nums1 = [1, 3]
nums2 = [2]

The median is 2.0

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

2 二分查找解题

这道题目如果不要求时间复杂度的话，很简单。把两个数组合并。如果长度为奇数，找到中间位置的下标返回该元素。如果长度是偶数，则找到中间两个元素，求平均值返回。要符合时间复杂度要求就要用二分。以下内容来源于力扣，只看解法四。

2.1中位数的定义

中位数将一个数组分成两部分，左边元素个数等于右边元素个数，并且左边元素的值都小于右边元素的值。

2.2 数组切分

一个长度为m的数组，可以有m+1个位置可以切。
在这里插入图片描述

我们把数组nums1，和nums2分别在位置i和j进行切割。
在这里插入图片描述
将nums1左边的元素和nums2左边的元素合并在一起称为left_part。将nums1右边的元素和num2右边的元素合并在一起称为right_part。

2.3 寻找成立的条件

如果nums1和nums2的长度为偶数，如果能够保证以下两个条件。
条件1：left_part的长度比right_part的长度多1，也就是说 $i + j = m - i + n - j + 1$ $i + j = (m + n + 1) / 2$
条件2：left_part的最大值<=right_part的最小值，也就是说max(nums1[i-1],nums2[j-1])<=min(nums1[i],nums2[j])

那么中位数可以表示为(left_part的最大值+right_part的最小值)/2

如果nums1和nums2的长度为奇数，如果能够保证以下两个条件。
条件1：left_part的长度比right_part的长度相同，也就是说 $i + j = m - i + n - j$ $i + j = (m + n) / 2$
条件2：left_part的最大值<=right_part的最小值，也就是说max(nums1[i-1],nums2[j-1])<=min(nums1[i],nums2[j])

那么中位数可以表示为left_part的最大值。

将奇数和偶数情况合并一下就是：
结果：长度为奇数：结果=left_part的最大值；否则，结果=(left_part的最大值+right_part的最小值)/2
条件1：i+j = (m+n+1)/2
条件2：max(nums1[i-1],nums2[j-1])<=min(nums1[i],nums2[j])，也就是说nums1[i-1]<=nums2[j] 并且 nums2[j-1]<=nums1[i]

对于nums1[i-1]>nums2[j] ,并且为了不越界，需要保证i!=0 并且 j!=n
在这里插入图片描述
这种情况下，需要减少i。因为条件1，j自然增加。

对于nums2[j-1]>nums1[i]，并且为了不越界，需要保证j!=0 并且 i!=m
在这里插入图片描述
这种情况下需要增加i。
接下来讨论边界情况。
当 i=0，或者j=0的时候，也就是切在了最左边。

当i=0，left_part的最大值=nums2[j-1]；当j=0，left_part的最大值=nums1[i-1]，right_part的最小值计算方法不变。

当i=m，或者j=n的时候，也就是切在了最右边。
在这里插入图片描述
当i=m的时候，right_part的最小值=nums2[j]；当j=n的时候，right_part的最小值=nums1[i]。left_part的最大值计算方法不变。

2.4二分查找合适的i

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int n1 = nums1.length;
        int n2 = nums2.length;
        if(n1>n2){
            return findMedianSortedArrays(nums2,nums1);
        }
        int k = (n1+n2+1)/2;
        int l = 0, r = n1;
        while(l<=r){
            int i  = l+((r-l)>>1);
            int j = k - i;
            if(i!=0 && j!=n2 && nums1[i-1]>nums2[j]){
                r = i -1;
            }else if(j!=0 && i!=n1 && nums2[j-1]>nums1[i]){
                l = i + 1;
            }else{
                int 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((n1+n2)%2==1){
                    return maxLeft;
                }
                int minRight = 0;
                if(i==n1){
                    minRight = nums2[j];
                }else if(j==n2){
                    minRight = nums1[i];
                }else{
                    minRight = Math.min(nums1[i],nums2[j]);
                }
                return (maxLeft+minRight)/2.0;
            }
        }
        
        return 0.0;
            
    }
}