leetCode-04 《两个正序数组的中位数》解析

两个正序数组的中位数

描述

给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?

示例1

输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2

示例2

输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5

示例3

输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000

示例4

输入:nums1 = [], nums2 = [1]
输出:1.00000

示例5

输入:nums1 = [2], nums2 = []
输出:2.00000

条件

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */

方法

1. 暴力法

比较好理解,就是合并数组,然后sort排序。
按题目要求,合并的数组长度是奇数时找出中间数,偶数时找出中间两个数并求平均数。
代码如下

var findMedianSortedArrays = function(nums1, nums2) {
  let allNums = nums1.concat(nums2)
  let sortAllNums = allNums.sort((a, b) => a - b) // 排序后的
  if (sortAllNums.length % 2) { // 奇数
    return sortAllNums[(sortAllNums.length - 1) / 2]
  } else { // 偶数
    return (sortAllNums[sortAllNums.length / 2] + sortAllNums[sortAllNums.length / 2 - 1]) / 2
  }

2. 二分法

本文主要讲二分法,因为题目的进阶要求是时间复杂度为 O(log (m+n))。这种时间复杂度有对数的时候,一般都要想到用二分法来处理这个问题。

首先,我们规定数组A的长度为m,数组B的长度为n,且m <= n。这里我们以小的数组做2分。

对于A数组,以l代表分割线位置,分割线左范围值l0 = 0置,分割线右范围值li = m。那么分割线位置i应该是 (l0 + li) / 2

那么数组A以 i 作二分,数组B以 j 作二分。那么根据题目的要求,i + j 要满足等于 (m + n + 1) / 2 (编程语言向下取整)。

20201107213428.jpg

那么所得的分各项两边的数需要满足如下条件:

  1. Ai < Bj+1

  2. Bj < Ai+1

满足上述条件,再根据m + n的奇偶性求出中位数。

如果不满足上述条件,说明分割线的位置错误。需要重新利用二分法二分剩下的位置,去寻找合适的分割线。

如果Ai > Bj+1 A数组分割线左边大了,说明分割线太靠右了,需要向左移动。将分割线右范围值在目前分割线的基础上向左移动一位。li = i - 1。

如果Bj > Ai+1 A数组右边小了,说明分割线太靠左了,需要向右移动,将分割线左范围在原来基础上向右移动一位。l0 = i + 1。

那么 i = (l0 + li) / 2; j = (m + n + 1) - i。

推理过程如下图:

过程图

过程图

代码

const findMedianSortedArrays  = (nums1, nums2) => {
  // 确保第一个数组长度较小
  if (nums1.length > nums2.length) {
    [nums1, nums2] = [nums2, nums1]
  }

  const m = nums1.length
  const n = nums2.length
  let low = 0
  let high = m

  while (low <= high) {
    const i = Math.floor((high + low) / 2)
    const j = Math.floor((m + n + 1) / 2) - i

    const maxLeftA = i === 0 ? -Infinity : nums1[i - 1]
    const minRightA = i === m ? Infinity : nums1[i]
    const maxLeftB = j === 0 ? -Infinity : nums2[j - 1]
    const minRightB = j === n ? Infinity : nums2[j]

    // 判断上面4个数的关系
    if (maxLeftA <= minRightB && minRightA >= maxLeftB) { // 满足条件
      // 判断奇偶性 然后得出结果
      return (m + n) % 2 === 1 ? Math.max(maxLeftA, maxLeftB) : (Math.max(maxLeftA, maxLeftB) + Math.min(minRightA, minRightB)) / 2
    }
    if (maxLeftA > minRightB) { // 分割线需要左移动
      high = i - 1
    }
    if (minRightA < maxLeftB) { // 分割线需要右移动
      low = i + 1
    }
  }
}

本文参考资料为leetCode的清华大佬lucifer

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