LeetCode04(寻找两个有序数组的中位数)

题目如下所示:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

示例 1:

nums1 = [1, 3]

nums2 = [2]

则中位数是 2.0

示例 2:

nums1 = [1, 2]

nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5


假设两个数组情况如下:

num[1]:    1    3    4    7

num[2]:    1     2    3    4    5    6    7    8    9    10

因为时间复杂度为log,则应该采用二分查找法。

题目中的要求是找到两数组的中位数,相当于“找到数组中第k位数”的一个特殊例子,也就是说现在两数组一共有14个元素,则中位数应该是第7小的数字。(K= 7)

思路如下:

        首先是找到第 7/2 = 3小的数,同时找到num1[3]和num2[3]的值,比较哪个值更大。上面可以看到是num1[3] > num2[3],则可以把num2中的num2[0] ~ num2[2]都可以排除掉。因为num1[3] > num2[3],即可以肯定num[1] > num2[0] ~ num2[2],所以这部分的值绝对不是我们所要找的“第7小的数”所在的位置。

        当排除num2[0] ~ num2[2]后,需要重新设置数组的起始指针,即下次寻找中,应该从num2[4]开始,这样才做到排除的效果,同理num1中也要这样操作。

        当排除3个不需要的数以后,在当前窗口中,需要寻找的是第k = 7 - 3 =4个最小的数。(即K值变化与所排除的数目有关)

        这道题采用了递归的方法,递归的结束条件为 K = 1的时候,返回两数组中更小的那个值。

        现在来介绍GetK函数。首先先定义了len1和len2,这两个值是表示当前num1和num2的窗口值(因为每次进行排除以后,会缩减窗口)。然后语句if (len1 > len2) return GetK(num2,start2,end2, num1, start1, end1 ,k); 是为了保持num1的元素数目永远小于num2,从而为了下一行语句打基础,if(len1 == 0) return num2[ start2 + k -1],即当前仅当num1中的窗口值为0,则可以直接在num2中寻找即可,这样做免除了判断num1或num2哪个窗口为空的两种情况,令代码更简洁。然后是更新num1和num2的头指针的位置,因为每次操作都会进行排除,即要把头指针移动到新的数组头位置,这里取决于k/2和len的大小(上文有提及)。这一些工作做好了以后,就来到关键部分,判断num1[i]和num2[j]的大小,选择排除哪部分。直到K=1的时候,递归结束并且返回正确值。

        特殊情况一,就是当k = 10时,即 k/2 = 5,超过了num1数组的元素总数量,所以num1应该取最后一个值与num2比较,这里需要代码比较。

        特殊情况二,如果两数组元素总和为奇数的时候,则只要让k = total /2即可,但是如果两数组元素总和为偶数的时候,要寻找两次,则 k = (total + 1) / 2 和 k = (total + 2) /2 ,两次获取的和再除以2才能获得答案。为了满足这个情况,无论是基数还是偶数,都计算k = (total + 1) / 2 和 k = (total + 2) /2,偶数的情况没有问题,奇数的时候K值都是一样的,只是计算多一次而已。

        除此以外,代码中还要一个小技巧,就是如果发现num1的长度 > num2的长度,则交换。这样就保证任何情况下都是num1会产生当前窗口为0的情况,这样就可以直接在num2中查找即可。



代码如下:

class Solution {

    public double findMedianSortedArrays(int[] nums1, int[] nums2) {

        int x = nums1.length;

        int y = nums2.length;

        int KsLeft = (x + y + 1) / 2;

        int Ksright = (x + y + 2) / 2;          //这两个变量是给计算总数为偶数的时候所使用的,即第k位和第k+1位相加再除2的结果。

        return (GetK(nums1, 0, x-1, nums2, 0 ,y-1 , KsLeft) + GetK(nums1, 0 , x-1, nums2, 0 ,y-1, Ksright)) * 0.5;

        }

    private int GetK(int [] num1, int start1, int end1 , int []num2, int start2, int end2, int k){

        int len1 = end1 - start1 + 1;           //计算当前修改过的数组的长度

        int len2 = end2 - start2 + 1;           

        if (len1 > len2) return GetK(num2,start2,end2, num1, start1, end1 ,k);      //总是让len1的长度小于len2

        if(len1 == 0) return num2[ start2 + k -1];                    //这里是一个小技巧,令1数组的长度小于2数组,这样只要1数组为空,就可以直接返回2数组的值


        if(k==1) return Math.min(num1[start1] , num2[start2]);            //最后一步,K=1的时候取两数组当前最小值的更小值

        int i = start1 + Math.min(len1 , k/2) - 1;

        int j = start2 + Math.min(len2 , k/2) - 1;                  //判断当前所需要的K值是否大于数组总长度

        if(num1[i] > num2[j]) return GetK(num1, start1, end1, num2, j + 1 , end2, k - (j - start2 + 1));

        else return GetK(num1, i + 1, end1 ,num2, start2, end2, k -( i - start1 +1));

    }

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 216,001评论 6 498
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,210评论 3 392
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 161,874评论 0 351
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,001评论 1 291
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,022评论 6 388
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,005评论 1 295
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,929评论 3 416
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,742评论 0 271
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,193评论 1 309
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,427评论 2 331
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,583评论 1 346
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,305评论 5 342
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,911评论 3 325
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,564评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,731评论 1 268
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,581评论 2 368
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,478评论 2 352

推荐阅读更多精彩内容