来自北大算法课的Leetcode题解:4. 寻找两个正序数组的中位数

代码仓库Github | Leetcode solutions @doubleZ0108 from Peking University.

  • 解法1: 最基础的想法肯定是引入一个新数组,把两个数组都放进来,然后排序,直接根据奇偶选中位数,但感觉肯定会超时或者爆显存
  • 解法2(T88% S30%): 因为题干强调了log的时间复杂度,所以肯定是二分查找的思想,通过两个指针分别指向两数组的开头,求解中位数本质上就是把前n/2-1个最小的数扔掉,这个思想可以很好的通过二分查找来实现,不过需要注意的是有可能一个数组很短而另一个很长,因此需要判断是否有指针已经到尾部,如果是的话只在另一个数组中移动就可以了。扔完n/2-1个数之后,根据原数组长度的奇偶性来选取中位数即可,一个小技巧,可以直接把之后的数组切片合并成一个数组然后讨论第0号和1号元素,这样省时方便又不用担心爆内存(因为至少有一半已经被排除掉了)
var findMedianSortedArrays = function(nums1, nums2) {
    var flag = (nums1.length + nums2.length) % 2 == 0;   // 中位数是中间两个数的平均
    var midNum = Math.floor((nums1.length + nums2.length)/2);
    for (var i=0, j=0; i+j<(flag ? midNum-1 : midNum); ) {
        if (i == nums1.length) {
            j++;
            continue;
        } else if (j == nums2.length) {
            i++;
            continue;
        }

        if (nums1[i] < nums2[j]) {
            i++;
        } else {
            j++;
        }
    }

    var tmp = [...nums1.slice(i), ...nums2.slice(j)].sort((a,b)=>a-b);
    if (!flag) {
        return tmp[0];
    } else {
        
        return (tmp[0]+tmp[1]) / 2;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

doubleZ0108

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值