LeetCode[4] - 两个排序数组的中位数

~~

好久没写一些乱七八糟的东西了,两个月前裸辞离开了大连,成功加入杭漂大家庭,很顺利的进入到了一家“独角兽”~~ 只面了这一家,拿了offer就在杭州玩了一个星期就入职了。上家拖欠的工资依旧没有结算。
来这主要负责基于socketio的IM部分功能开发~~ 最近开发任务不是很重就刷刷leetcode,解闷······
如果有怎么不对的地方或者其他想法可以一起讨论哟 ~~ 楼主玻璃心 勿喷的太严重~~

正文

之前一直刷的都是简单、中等,突然发现个困难level的还有点小激动。
题目是这样的:



乍一看很简单,当然最容易想到的解法就是两个数组合并成一个然后找结果~但是既然是算法题那是否有更优的选择呢?

心路历程

首先,根据这两个数组我们能获取到的已知信息:

  • 数组总长度:size=nums1.length+nums2.length
  • 中位数的位置:size/2或者size/2,size/2-1(后者需要取平均数)
  • 是否需要取平均数:size % 2 == 0
    这是目前比较直观的信息。
    假如人工去处理这个问题会怎么个过程?
  • 拿到两者第一项比较
  • 选择较小的一方,拿较小一方的下一项和较大一方的第一项比较
  • 选择较小的一方,拿较小一方的下一项和较大一方的当前项比较
  • 递归中···当取得size/2项停止递归
  • 若有一方数组无下一项,那直接在另一方数组中查找
    思路有了那么代码就是翻译的过程····
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int size = nums1.length + nums2.length;//数组总长度
        int bingo = size / 2;//中位数所在项
        int index1 = 0;//nums1遍历下标位置
        int index2 = 0;//nums2遍历下标位置
        double pre = 0;//当前结果前一项,用于算平均数
        double now = 0;//当前结果
        int[] currentArray;//当前遍历数组
        int currentIndex;//当前遍历下标
        if (size == 0) {
            return 0;
        }
        for (int i = 0; i <= bingo; i++) {
            pre = now;
            if (nums1.length <= index1) {//nums1无下一项,直接从nums2中取中位数
                int k = index2 + bingo - i;
                now = nums2[k];
                if (k > 0 && pre < nums2[k - 1]) {//pre在nums1中或在nums2中
                    pre = nums2[k - 1];
                }
                break;
            } else if (nums2.length <= index2) {//同上
                int k = index1 + bingo - i;
                now = nums1[k];
                if (k > 0 && pre < nums1[k - 1]) {
                    pre = nums1[k - 1];
                }
                break;
            }
            if (nums1[index1] <= nums2[index2]) {//选取当前项中较小的数组,下标后移
                currentArray = nums1;
                currentIndex = index1;
                index1++;
            } else {
                currentArray = nums2;
                currentIndex = index2;
                index2++;
            }
            now = currentArray[currentIndex];
        }
        return size % 2 == 0 ? (now + pre) / 2 : now;
    }

因为智商不够只能写写“面向过程”算法,但是做出来还是蛮有成就感的~~

运行
运行结果统计

抱着学习的心态,我复制了速度排行第一的源码,很简洁:

    public double findMedianSortedArrays1(int[] nums1, int[] nums2) {
        int m = nums1.length, n = nums2.length, left = (m + n + 1) / 2, right = (m + n + 2) / 2;
        double res = (findKth(nums1, nums2, left) + findKth(nums1, nums2, right)) / 2.0;
        return res;
    }

    int findKth(int[] nums1, int[] nums2, int k) {
        int m = nums1.length, n = nums2.length;
        if (m > n) return findKth(nums2, nums1, k);
        if (m == 0) return nums2[k - 1];
        if (k == 1) return Math.min(nums1[0], nums2[0]);
        int i = Math.min(m, k / 2), j = Math.min(n, k / 2);
        if (nums1[i - 1] > nums2[j - 1]) {
            return findKth(nums1, Arrays.copyOfRange(nums2, j, n), k - j);
        } else {
            return findKth(Arrays.copyOfRange(nums1, i, m), nums2, k - i);
        }

一看就是递归大佬,但递归的效率应该是低于遍历的吧,而且又有大量的Arrays.copyOfRange于是我打算搞点测试用例pk一下,size为数组长度,max为数组项最大值,数组项随机生成:

测试用例:size,max 递归(ms) 非递归(ms)
nums1:10000000,10000; nums2: 100,10000 562 22
nums1:10000000,10000000; nums2: 10000000,10000000 619 63
nums1:10000000,100; nums2: 10000000,10 505 29

单纯从时间上来看果然递归效率并不是特别好,还有很多测试用例我就不放表格里了,简书的markdown真的是有待完善···

总结

这么来看leetcode的统计也不是特别准确,仅供参考吧。这道算法题难度定位可能也不是特别准确,之前做的简单level难度上确实有超过这个的,可能每个人擅长和思考方式不一样吧。先写到这,工作啦~
按照时间复杂度来计算 后者却是完全符合题意。我的时间复杂度稍微高一点…

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

推荐阅读更多精彩内容