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

题目描述

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

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

则中位数是 2.0

C++代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) { 
        int m = nums1.size();
        int n = nums2.size();
        if (m > n){
            return findMedianSortedArrays(nums2, nums1);
        }
        int low = 0, high = m;
        while(low <= high){
            int i = (low + high) / 2;
            int j = (m + n + 1) / 2 - i;
            if (i > low && nums1[i-1] > nums2[j])
                high = i - 1;
            else if (i < high && nums2[j-1] > nums1[i])
                low = i + 1;
            else{
                float maxLeft = 0;
                if (i == 0) maxLeft = nums2[j-1];
                else if (j == 0) maxLeft = nums1[i-1];
                else maxLeft = max(nums1[i-1], nums2[j-1]);
                if((m+n) & 1 == 1) return maxLeft;

                float maxRight = 0;
                if(j == n) maxRight = nums1[i];
                else if (i == m) maxRight = nums2[j];
                else maxRight = min(nums1[i], nums2[j]);
                return (maxLeft + maxRight) / 2;
            }
        }
        return 0;
    }
};

题目思路

在一个集合中中位数起到一个划分的作用，它把这个集合划分为两个集合，其中一个集合中的任一元素小于另一集合中的任一元素。
在两个有序的数组中(nums1的长度为m，nums2的长度为n)，我们分别在nums1的下标为i的元素前插入一个隔板，在nums2的下标为j的元素前插入隔板。

Left	Right
nums1[0] nums1[1] … nums[i-1]	nums1[i] nums1[i+1] … nums1[m]
nums2[0] nums2[1] … nums2[j-1]	nums2[j] nums[j+1] … nums2[n]

只要满足Left.len == Right.len(或者Left.len = Right.len + 1),并且Max(Left) <= Min(Right)我们就可以找到中位数。
即i + j = m - i + n - j (或m - i + n - j + 1) &&
max(nums1[i-1], nums2[j-1]) <= min(nums1[i], nums2[j])
由1式我们可以得出j = (m + n + 1) / 2 - i,因为是int型的计算，所以两个1式都可以满足。因为0 <= i <= m,而j = (m+n+1)/2 -i,为了保证j是一个合法的数，我们需要让m<=n。
由2式可得 nums1[i-1] <= nums2[j] && nums2[j-1] <= nums1[i]。

接下来我们可以按照以下步骤来进行二分搜索

1. 令low = 0,high = m,在nums1中进行二分搜索
2. 令i = (low + high) / 2, j = (m+n+1) / 2 - i
3. 我们可以得到三种情况
   • nums1[i-1] > nums2[j]
     这种情况说明了i太大了需要缩小i，则令high = i - 1,继续二分
   • nums2[j-1] > nums1[i]
     这种情况说明了i太小了需要增大i，令low = i + 1，继续二分
   • nums1[i-1] <= nums2[j] && nums2[j-1] <= nums1[i]
     说明了我们找到了正确的i

我们取左边的最大值为maxLeft，右边的最小值为minRight。当m+n是个奇数的时候，中位数就是maxLeft；m+n是个偶数的时候，中位数是(maxLeft + minRight) / 2,。接下来我们就需要考虑如何得到这个maxLeft和minRight。

• 当i == 0的时候，说明nums1的所有元素都在Right，那么maxLeft就是nums2中隔板前的那个元素，也就是nums2[j-1]

• 当j== 0的时候，说明nums2的所有元素都在Right，那么maxLeft就是nums1中隔板前的那个元素，也就是nums1[i-1]

• 不然的话，maxLeft = max(nums1[i-1], nums2[j-1])

• 当i == m的时候，说明nums1的所有元素都在Left，那么minRight就是nums2中隔板后的那个元素，也就是nums2[j]

• 当j == n的时候，说明nums2的所有元素都在Left，那么minRight就是nums1中隔板后的那个元素，也就是nums2[i]

• 不然的话，minRight = min(nums[i], nums[j])