LeetCode题：寻找两个有序数组的中位数

题目描述：

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
 

思路描述

对于一个有n个数的数组，其中位数可以看作第(n+1)/2小和第(n+2)/2小的两个数的平均数。于是问题就转化为在两个数组中分别找出第(n+1)/2小和第(n+2)/2小的数。

为了找到两个数组中第k小的数，可以分别把nums1和nums2分成两个部分，使两个数组左边部分的值都不大于右边部分的值。nums1两部分分别有l1,r1个数，nums2两部分分别有l2,r2个数。对于两个分界线的位置，可以分别用两个整数p1，p2来表示，其中p1=l1,p2=l2（把p看成分割数组一条线，则这条线的左边有p个数，显然p的取值最小为0，最大为该数组元素个数）。我们希望找出一对l1,l2,使l1+l2=k，此时p1+p2也等于k，这样第k小的数就在两个分界线的左边的数中的最大的那个。因为两个数组是有序递增的，所以这个最大数（即第k小的数）是最靠近分界线的两个数中较大的那个（这题描述中只提到有序，所以可能也包括递减，但事实上假设递增也不影响ac）。

所以我们可以对nums1的分界线位置用二分法，初始l=0（对应分界线在nums1数组最左边），r=m（对应分界线在nums1数组最右边，p=(l+r)/2,循环中根据对应情况调整l，r。直到出现两个数组左边部分的值都不大于右边部分的值。

用上述思路找出第（m+n+1）小的数和第（m+n+2）小的数再取平均即得最终答案。

 

时间空间效率：

 

C++代码如下：

class Solution {
public:
    double findNumOfTwoArrays(vector<int>& nums1, vector<int>& nums2, int k){
        //第k小的数，nums1长度大于等于nums2
        if (nums1.size()==0){
            return nums2[k-1];
        }
        int n1=nums1.size(),n2=nums2.size();
        int l = 0,r = n1,leftNum,leftNum2;
        bool flag=false;
        while (l<=r){
            leftNum = (l+r)/2;
            leftNum2 = k - leftNum;
            if (leftNum2<0){
                r = leftNum-1;
                continue;
            }
            if (leftNum2>n2){
                l = leftNum+1;
                continue;
            }   
            if (leftNum>=1&&leftNum<=n1-1&&leftNum2>=1&&leftNum2<=n2-1){
                if (nums1[leftNum-1]<=nums2[leftNum2]&&nums1[leftNum]>=nums2[leftNum2-1]){
                    return max(nums1[leftNum-1],nums2[leftNum2-1]);
                }
                if (nums1[leftNum-1]>nums2[leftNum2]){
                	r = leftNum-1;
				}
				else{
					l = leftNum+1;
				}
				continue;
            }
            if (leftNum>=1&&leftNum<=n1-1&&leftNum2==0){
                if (nums1[leftNum-1]<=nums2[leftNum2]){
                    return nums1[leftNum-1];
                }
                r = leftNum-1;
                continue;
            }
            if (leftNum>=1&&leftNum<=n1-1&&leftNum2==n2){
                if (nums1[leftNum]>=nums2[leftNum2-1]){
                    return max(nums1[leftNum-1],nums2[leftNum2-1]);
                }
                l = leftNum+1;
                continue;
            }
            if (leftNum==0&&leftNum2>=1&&leftNum2<=n2-1){
                if (nums1[leftNum]>=nums2[leftNum2-1]){
                    return nums2[leftNum2-1];
                }
                l = leftNum+1;
                continue;
            }
            if (leftNum==0&&leftNum2==n2){
                if (nums1[leftNum]>=nums2[leftNum2-1]){
                    return nums2[leftNum2-1];
                }
                l = leftNum+1;
                continue;
            }
            if (leftNum==n1&&leftNum2>=1&&leftNum2<=n2-1){
                if (nums1[leftNum-1]<=nums2[leftNum2]){
                    return max(nums1[leftNum-1],nums2[leftNum2-1]);
                }
                r = leftNum - 1;
                continue;
            }
            if (leftNum==n1&&leftNum2==0){
                if (nums1[leftNum-1]<=nums2[leftNum2]){
                    return nums1[leftNum-1];
                }
                r = leftNum - 1;
                continue;
            }
            if (leftNum==n1&&leftNum2==n2){
                return max(nums1[leftNum-1],nums2[leftNum2-1]);
            }
        }
        if (leftNum<=0)
            return nums2[k-1];
        return nums2[k-n1-1];
    }
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        if (nums1.size()<nums2.size()){
            nums1.swap(nums2);
        }
        int n1=nums1.size(),n2=nums2.size();
        if (n2==0) return (double)(nums1[(n1+1)/2-1]+nums1[(n1+2)/2-1])/2.0;
        int ans1,ans2;
        ans1 = findNumOfTwoArrays(nums1,nums2,(n1+n2+1)/2);
        ans2 = findNumOfTwoArrays(nums1,nums2,(n1+n2+2)/2);
        return (ans1+ans2)/2.0;
    }
};