leetcode--P4 Median of Two Sorted Arrays

There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

这个题目的所求意思不难，给你两个升序排序的数组然后求出他们两者的一个中位数，但最后要求时间复杂度为log级别的，这就有点难办了，连让我先用O（n）复杂度错一次的机会都不给了。

在尝试一些方法未果后，百度了一下网上大牛们的做法，基本和自己想的一致，不过对于第k大的值的处理上聪明很多。那么先来说一下这题的一个思路：

所谓求两个数组的中位数，就是求第（m+n）/2大的一个值，即传化成了求第k大值的一个问题。那么我们先不考虑一些边界条件，假设这两个数组分别为a,b，并且他们的大小都大于k/2，那么我们去比较a[k/2-1]和b[k/2-1]的大小，那么出现以下三种情况：

如果a[k/2-1]<b[k/2-1]，这表示a[0]到a[k/2-1]的元素都在a和b合并之后的前k小的元素中。换句话说，a[k/2-1]不可能大于两数组合并之后的第k小值，所以我们可以将其抛弃；

那么当a[k/2-1]>b[k/2-1]可以得出类似的推论，并做相应的操作；

当a[k/2-1]=b[k/2-1]时，我们已经找到了第k小的数，也即这个相等的元素，我们将其记为m。由于在a和b中分别有k/2-1个元素小于m，所以m即是第k小的数。

那么这一题的大致流程也就出来了，代码如下：

<span style="font-size:14px;">int min(int x,int y)
{
    if(x<y) return x;
    else return y;
}

double findKth(int a[], int m, int b[], int n, int k)
{
    if(n<m)
        return findKth(b,n,a,m,k);
    if(m==0)
        return b[k-1];
    if(k==1)
        return min(a[0],b[0]);
    int pa = min(m,k/2);
    int pb = k - pa;
    if(a[pa-1] < b[pb-1])
        return findKth(a+pa,m-pa,b,n,k-pa);
    if(a[pa-1] > b[pb-1])
        return findKth(a,m,b+pb,n-pb,k-pb);
    return a[pa-1];
}

double findMedianSortedArrays(int A[], int m, int B[], int n)
{
    // Note: The Solution object is instantiated only once.
    int total = m+n;
    if(total%2!=0)
        return findKth(A,m,B,n,total/2+1);
    else
        return (findKth(A,m,B,n,total/2)+findKth(A,m,B,n,total/2+1))/2;
}</span>

时间上由于参考了网上大神的优化方法，故比较不错为24ms。