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

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

题目链接

中位数

题目

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数 。算法的时间复杂度应该为 O(log (m+n)) 。
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-10⁶ = nums1[i], nums2[i] <= 10⁶

输入输出示例

示例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

思路

题目对时间复杂度有要求，常规想法先将两个数组合并排列再找中位数时间复杂度为 O((m+n)log(m+n))，不能满足时间要求。
利用所给数组是正序数组的性质，同时可知中位数左边的数小于等于中位数，右边的大于等于它，如下图，用边界 i、j 可将两个数组划分成左边一块右边一块，m+n为偶数时就划分为左右个数相等的部分，m+n为奇数则划分左边比右边多一个，即可推导出 i + j =（m+n+1)/2，同时这个边界需要满足的要求是左边最大的数要小于等于右边最小的数。如此可以看出当m+n为奇数时，中位数就是左边部分最大的那个数，m+n为偶数时，中位数为左边最大的数加右边最小的数的二分之一。由此找到 i 符合的值即可确定边界，找到中位数，i 的值可以用二分法来确定。

还要注意一些特殊情况，例如 i = 0时左边最大值即nums2[ j - 1 ]，i = nums1Size时，右边最小值即nums2[ j ]。

参考代码

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
	//判断两个数组大小，保证nums1Size小于等于nums2Size
	if(nums1Size>nums2Size) return findMedianSortedArrays(nums2,nums2Size,nums1,nums1Size);
	int i,j,left=0,right = nums1Size;//初始化i的查找区间
    if(left<=right)
    {
        while(left<=right)
        {
            i = (left+right)/2;//二分法
            j = (nums1Size+nums2Size+1)/2 - i;
            if(j!=0&&i!=nums1Size&&nums2[j-1]>nums1[i]) left = i+1;//判断i是否需要且能增大	
            else if(i!=0&&j!=nums2Size&&nums1[i-1]>nums2[j]) right = i-1;//判断i是否需要且能减小
            else
            {
                int max,min;
                if(i==0) max = nums2[j-1];//特殊情况
                else if(j==0) max = nums1[i-1];
                else max = nums1[i-1]>nums2[j-1]?nums1[i-1]:nums2[j-1];
                if((nums1Size+nums2Size)%2==1) return max;//奇数时直接返回即可
                if(i==nums1Size) min = nums2[j];
                else if(j==nums2Size) min = nums1[i];
                else min = nums1[i]<nums2[j]?nums1[i]:nums2[j];
                return (max+min)/2.0;
            }
        }
        return 0;
    } 
    else return 0;
}