leetcode 4 寻找两个正序数组的中位数（二分法，第K个最小的数）

题目描述：
给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

思路：
考虑到时间复杂度是log(m+n),因此使用二分法来计算
例如一组数1234567，求其中位数，相当于求第4小的数
因此本题目求中位数也相当于求第K小的数
考虑到数的个数是奇数还是偶数
因此求第（m+n+1)/2小和（m+n+2)/2小的数的平均值
（如果是偶数个，则（m+n+1)/2和m+n+2)/2是一样的数

接下来就是写一个用二分法求第K小的数的函数
每次排除k/2个数

原文戳这里

代码如下：

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m=nums1.size();
        int n=nums2.size();
        int left=(m+n+1)/2;
        int right=(m+n+2)/2;
        return (getKth(nums1,0,m-1,nums2,0,n-1,left)+getKth(nums1,0,m-1,nums2,0,n-1,right))/2.0;
    }
    int getKth(vector<int>nums1,int start1,int end1,vector<int>nums2,int start2,int end2,int k){
        int len1=end1-start1+1;
        int len2=end2-start2+1;
        if(len1>len2) return getKth(nums2,start2,end2,nums1,start1,end1,k);
        if(len1==0) return nums2[start2+k-1];
        if(k==1) return min(nums1[start1],nums2[start2]);
        int i=start1+min(len1,k/2)-1;
        int j=start2+min(len2,k/2)-1;
        if(nums1[i]>nums2[j]){
            return getKth(nums1,start1,end1,nums2,j+1,end2,k-(j-start2+1));
        }
        else return getKth(nums1,i+1,end1,nums2,start2,end2,k-(i-start1+1));
    }
};