计算两个有序数组的中位数（C++）

题目描述： 

计算两个有序数组的中位数，至少一个数组不为空。要求时间复杂度O(log(m+n))，m、n分别为两个数组的长度。

如：

     {1,2} {3} 输出应为2

     {1,3} {2,5} 输出应为2.5

题目分析：

        时间复杂度要求为log级别，数组又都是有序的，容易联想到二分，每次比较去掉一部分元素。

        数组中位数第一反应是需要分奇偶。奇数的话，我们只需要找到两个数组长度的中间那个下标元素即可，偶数则需要求个平均，因此问题的关键转化为了查找两数组的第k小元素。

        第一个数组aa，第二个数组bb。找第k小元素，需要分别在两个数组中找到某个元素（第a、b个元素），满足a+b=k，且aa[a]>=bb[b-1]、bb[b]>=aa[a-1]。我们假设第一个数组长度小于第二个数组（否则交换下），如果前一个数组为空，那直接去后者的第k个元素即可；如果k为1（k不可能为0，因为题目规定至少一个数组不为空），取两数组第一个元素最小值即可（第一个数组长度小于第二个数组，因此第一个数组不为空，后者也不会为空）；比较aa中第k/2的元素和满足取k个元素的bb中对应的元素，较小的元素对应的数组的左半部分即可舍去（因为它们肯定是前k-1小中的元素），舍去这a(或b)个元素的同时，我们的寻找目标就变成了再剩余数组中找第k-a(或k-b)小元素。如此递归。

参考代码：

#include<vector>
#include<iostream>
using namespace std;

//计算两个已排好序的数组的中位数
double findMedianSortedArrays(vector<int>& a, vector<int>& b);

int main()
{
    int a[] = {1,2};
    int b[] = {3,5,5,5};
    vector<int> nums1(a,a+sizeof(a)/sizeof(int));
    vector<int> nums2(b,b+sizeof(b)/sizeof(int));
    cout<<findMedianSortedArrays(nums1,nums2)<<endl;

    return 0;
}

int findKth(vector<int>& aa, vector<int>& bb, int k)
{
    int len1 = aa.size();
    int len2 = bb.size();
    
    if(len1>len2)
        return findKth(bb,aa,k);
    
    //注意：先判断aa是否为空
    if(len1==0)
        return bb[k-1];
    if(k==1)
        return min(aa[0],bb[0]);
    
    int a = min(k/2,len1);
    int b = k-a;
    
    if(aa[a-1]<bb[b-1])
    {
        vector<int> m(aa.begin()+a,aa.end());
        return findKth(m,bb,k-a);
    }
    else if(aa[a-1]>bb[b-1])
    {
        vector<int> m(bb.begin()+b,bb.end());
        return findKth(aa,m,k-b);
    }
    else
        return aa[a-1];
}

double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2)
{
    int len1 = nums1.size();
    int len2 = nums2.size();
    
    int k = (len1+len2)/2;
    
    if((len1+len2)%2)
        return findKth(nums1,nums2,k+1);
    else
        return (findKth(nums1,nums2,k+1)+findKth(nums1,nums2,k))/2.0;
}