2020-02-05

int findKthEle( vector<int> & nums1, vector<int> & nums2, int k )
{
    int first = max( 0, k - nums2.size() ), last = min( nums1.size(), k );
    /* 我们寻找的是满足a[i-1]<=b[j]&&b[j-1]<=a[i]的最小的位置i */
    while ( first < last )
    {
        int mid = first + (last - first) / 2;
        /*
         * m=i,j=k-m,所以i+j=k,k-m-1相当于j-1即j前一个位置。j-1比i位置的值大,需要排除
         * 只需判断 b[j-1]<=a[i] 这个条件
         * 因为找的是最小的i,j=k-i是所有满足b[j-1]<=a[i]中相对最大的
         * 如果它不满足其他的也不可能满足另一个条件
         * 所以一定满足a[i-1]<=b[j],不满足的都会在逼近过程中排除
         */
        if ( nums2[k - mid - 1] > nums1[mid] )
            first = mid + 1;
        else last = mid;
    }

    /*
     * 循环结束时的位置first即为所求位置
     * 第k小即为max(nums1[first-1]),nums2[k-first-1])
     * 但是由于le可以为0、k,first-1或者k-first-1可能不存在
     */
    int nums1_max   = first == 0 ? INT_MIN : nums1[first - 1];
    int nums2_max   = first == k ? INT_MIN : nums2[k - first - 1];
    return(max( nums1_max, nums2_max ) );
}

double findMedianSortedArrays( vector<int> & nums1, vector<int> & nums2 )
{
    int n = nums1.size() + nums2.size();
    double  median;
    if ( n & 1 )    /* 奇数 */
    {
        median = findKthEle( nums1, nums2, n / 2 + 1 );
    } else {        /* 偶数 */
        median = (findKthEle( nums1, nums2, n / 2 ) + findKthEle( nums1, nums2, n / 2 + 1 ) ) / 2.0;
    }

    return(median);
}
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 至少在这里我是有衣服可穿的, 你不知我是谁, 我自然可以畅所欲言。
    一只四九阅读 226评论 0 0
  • 第八次接孩子放学。那天接到老师的电话,说孩子病了,我接了回来,结果,感恩爱人的做法,谢谢你。
    兔子的幸福生活阅读 196评论 0 5
  • 在iOS平台上想要实现不同App之间的内容分享的所有方式: 通过AirDrop、蓝牙、WiFi等实现不同设备的Ap...
    _森宇_阅读 2,315评论 0 0
  • 伊,是你 长长黑发浓密 每一根发丝 生在吾心血管里 缕缕相思都付与 伊人,晨雾星光都聚集眼底 濛濛,闪闪扑着流萤 ...
    云里风铮阅读 728评论 -1 18
  • 从淘宝用优惠券买了安记咖喱,安记咖喱在超市10块一盒,用优惠券19.6四盒。早些年喜欢吃德克士的咖喱鸡排饭,很多很...
    爱生活爱理财阅读 393评论 1 1