求两个数组的中位数

如题
这里写图片描述
这个题目难在时间复杂度上 ,要求O(log(m+n))。
很容易想到时间复杂度是O(m+n)的方法的:把2个数组合起来到一个数组,然后取中位数。
我们主要来说一下O(log(m+n))的算法。
首先中位数这个东西,如果 数的个数有奇数个和有偶数个 中位数取法是不一样的

所以我们 可以借鉴一下Manacher算法
都给它整成偶数啊
这个Manacher算法有时间讲下好了
反正我下次再讲就周五 了
这里是两个有序的数组
所以两个有序的数组都给它变成固定的奇数个数
就是虚拟加入“#”
注意是虚拟加 这样做起来就很方便了
所以两个数组总和就是2m+2n+2
假设第一个是m个原来 第二个是n个
这样两个有序数组合并以后的中位数就是m+n+1
这有什么好处呢,为什么这么加?因为这么加完之后,每个位置可以通过/2得到原来元素的位置。
这里写图片描述
这里写图片描述
解决了第一个问题

然后 接下来就是 怎么在规定时间内找中位数
这个时间复杂度 用的是二分法
接下来比较难理解
首先将两个数组二分
这里写图片描述
那我怎样从这个双数组里面取出第m+n+1个元素
ai < ai+1是肯定的
bj < bj+1也是肯定的
如果我们让ai<=bj+1&& bj+1<=ai+1
那就很巧了
然后如果前ai个元素和前bj个元素相加起来还等于m+n+1
那么第m+n+1个元素就是Max(ai,bj)
双击查看原图
但是如果这个m+n+1位置是一个“#”
那我们就要找它两边的元素做中位数了
所以把2个数组看做一个虚拟的数组A,目前有2m+2n+2个元素,割在m+n+1处,所以我们只需找到m+n+1位置的元素和m+n+2位置的元素就行了。
然后要是不巧的话
说明这个条件它不满足
这里写图片描述
这里就要二分啦
考虑时间复杂度所以 用短的二分
比较ai,bj和ai+1,bj+1。
假设A数组比较短
ai>bj+1,把amiddle减小,bmiddle增大。—> amiddle向左二分
bj+1>ai,把amiddle增大,bmiddle减小。—> amiddle向右二分
这样就可以log(m+n)了
代码:
这里写图片描述
正确率:
这里写图片描述

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值