【JS攻略Leetcode】No.4. median-of-two-sorted-arrays(两个排序数组的中位数)

引言:用Js攻略leetcode中的算法,将会介绍自己的思路和注意点,一边学习一边愉快刷题呀。

问题:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
你可以假设 nums1 和 nums2 不同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
中位数是 (2 + 3)/2 = 2.5

思考:

总体思考:
最主要的难点是算法复杂度的限制, log级别的时间复杂度我们自然想到二分法,每次取中间值,自然得出的是log2(n)级别的。
因为nums1和nums2是有序数组,我们对这两个数组分别在某一位置进行划分。

比如nums1数组设为A, 划分位置为 i :
left_A | right_A
A [0],A [1],...,A [i-1] | A [i],A [i + 1],...,A [m-1]
这样左边有 i 个元素,右边有( m - i )个元素。

nums2数组设为B,划分位置为 j :
left_B | right_B
B [0],B [1],...,B [j-1] | B [j],B [j + 1],...,B [n-1]
这样左边有 j 个元素,右边有( n - j )个元素。

把两个数组的左右两边分别混合,也就是left_A和left_B放一起,right_A和right_B放一起,得到中位数需要满足下面两个条件:

  1. 左右数量相等(偶数:i + j = m - i + n -j )或者左边比右边大一个(奇数:i + j = m - i + n -j + 1)
  2. max(left_A) 小于 min(right_B),而且max(left_B) 小于 min(right_A),因为left_A肯定小于right_A, left_B肯定小于right_B

那么对于总长度为奇数,中位数 = max(left_A, left_B);对于总长度为偶数,中位数 = ( max(left_A, left_B) ,min(right_A,right_B) ) / 2

把上面两个条件进行整理:

  1. i = 0〜m,j = parseInt( (m + n + 1) / 2) - i (奇偶通用,阴影划分的为左边,其他为右边)


    总长度为奇数

    总长度为偶数

    所以n要大于等于m(n>=m),这样确保 j 是合法索引。

  2. shortArray[ i -1] <= longArray[ j ],longArray[ j - 1 ] <= shortArray [ i ]

实现细节:

  1. 先对nums1和nums2的长度进行判断,选择长的作为longArray, 短的作为shortArray
  2. i 在 [ 0, m ]中进行取值,并根据二分法取中值,j = parseInt( (m + n + 1) / 2) - i
  3. 如果( j > 0 && i < m && longArray[j-1] > shortArray[i] ),说明shortArray [i]太小,i 应该继续增大,取值范围变为 [ i + 1, m ]
  4. 如果( i > 0 && j < n && shortArray[i-1] > longArray[j] ),说明shortArray [i]太大, i 应该减小,取值范围变为[0, i - 1]
  5. 以此循环,直到上面条件都不满足,说明要么取到边界了(i =0 , j=0 等情况),要么shortArray和longArray左边都取了值,要选出shortArray和longArray左边最大的值。
  • (i =0 情况下,左侧确定,左边最大值为longArray[j - 1] )


    image.png
  • (j = 0情况下,左侧确定,左侧最大值为shortArray[i - 1])


    image.png
  • (其他情况,shortArray和longArray左边都取了值)


    image.png
  1. 对于总长度为奇数, 中位数 = max(left_A, left_B), 也就是第5步得到的值。如果是奇数,要再得到 minRight.

代码:

var findMedianSortedArrays = function(nums1, nums2) {
    var shortArray = [],longArray = [], m = nums1.length, n = nums2.length,
         temp = 0, maxLeft = 0, minRight = 0;
    //保证 n >= m 
    if(m > n) {
        shortArray = nums2;
        longArray = nums1;
        temp = m;
        m = n;
        n = temp;
    } else {
        shortArray = nums1;
        longArray = nums2;
    }
    
    //假设shortArray取i个,longArray取parseInt((m+n+1)/2)-i个
    var imin = 0, imax = m, i, j;
    do {
        i = parseInt((imin + imax)/2);
        j = parseInt((m + n + 1)/2 - i);
        if(j > 0 && i < m && longArray[j-1] > shortArray[i]) {
            // shortArray和longArray左右都取了值,且i值取小了
            imin = i + 1;
        } else if( i > 0 && j < n && shortArray[i-1] > longArray[j]) {
            // shortArray和longArray左右都取了值,且i值取大了
            imax = i - 1;
        } else {
            // 考虑边界情况
            if(i == 0) {
                maxLeft = longArray[j - 1];
            } else if (j == 0 ) {
                maxLeft = shortArray[i - 1];
            } else {
                // i值不小不大, 也就是(shortArray[i-1] <= longArray[j] && longArray[j-1] <= shortArray[i])
                maxLeft = Math.max(shortArray[i - 1], longArray[j -1]);
            }
            break;
        }
    } while (imin <= imax);
    //两数组相加个数是奇数,只要取中间值
    if( (m + n)%2 == 1 ) return maxLeft;
    //个数是偶数,必须左右两边中位数相加除以二
    if(i == m) {
        minRight = longArray[j];
    } else if(j == n){
        minRight = shortArray[i];
    } else {
        minRight = Math.min(shortArray[i], longArray[j]);
    };
    return (maxLeft + minRight)/2;
}
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,723评论 6 481
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,485评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 152,998评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,323评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,355评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,079评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,389评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,019评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,519评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,971评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,100评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,738评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,293评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,289评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,517评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,547评论 2 354
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,834评论 2 345

推荐阅读更多精彩内容