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]


那么当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小的数。


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


int min(int x,int y)
{
    if(x 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;
}


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

你可能感兴趣的:(LeetCode)