Leetcode-4. Median of Two Sorted Arrays


Description
There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

Example 1:
nums1 = [1, 3]
nums2 = [2]
The median is 2.0

Example 2:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
分析
给定两个以排序的数组,要求输出这两个数组组成的新数组的中位数。
如[1,2]和[3],其中位数为2.[1,2]和[3,4]中位数为(2+3)/2。题目的难点在于要求复杂度不大于O(log (m+n))。

中位数:中位数是指一个可以将一系列数字串分为长度相等,并且中位数左边的数均不大于右边的数。
根据中位数的这个性质,我们只要找到一个数,使得两边的数子个数相同,并且左边的不大于右边的数,即可完成任务。

思路
由于给定的两个数组元素相对大小并不确定,因此数组元素直接拼接成一个数组并不实际。
我们可以将这两个数组分别划分,将其分成左右两部分,两个数组左边的部分构成一个新数组,右边的构成一个新数组。若左边的元素个数正好等于右边的元素个数,并且左边的最大数小于等于右边的最小数。这样我们就找到了中位数。
设较长的数组为pt1,长度为n,较短的数组为pt2,长度为m,两个整数i,j将这两个数组分为两部分。
设pt1分割后左边最大值为l1,右边最小值为r1,pt2分割后左边最大值为l2,右边最小值为r2。
若l1<=r2 && l2<=r1,则此时满足中位的条件。
中位数等于 [max(l1,l2)+min(r1,r2)]/2。

若l1>r2,则需要减小l1的值,即减小i的值。
若l2>r1,则需要减少l2的值,因为i+j保持不变,j减小,i必定增加。
i与j的几种边界条件:

  • i==0,此时短数组整体大于长数组,中位数位于长数组中,且i与j可能是正好需要的分割,但l2不存在(越界),因此将l2赋为最小值,保证判断正确。
  • j==0,此时长数组整体大于短数组,中位数位于长数组中,且i与j可能是所求分割。
  • i==2*m,此时短数组整体小于长数组,中位数位于长数组中,i与j可能符合要求,将r2赋最大值保证判断正确。
  • j=2*n,此时长数组整体小于短数组,中位数位于长数组中,i与j可能符合条件。

分割
对于数组进行分割,我们一般需要根据数组长度分奇数,偶数两种情况处理,比较麻烦。简单的方法是我们可以通过虚拟添加'#'符号的方式是得所有数组的长度均为奇数,便于处理。

有趣的是这样添加虚拟'#'符号以后,假设分割位置为i,则分割后

l=pt[(i-1)/2];
r=pt[i/2];

C语言代码

#include <stdio.h>
#include <string.h>
#include <stdlib.h>

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) 
{
    int n,m,i,j,imax,imin,temp,l1,l2,r1,r2;
    n=nums1Size;
    m=nums2Size;            //较短数组 
    int *pt1,*pt2;
    int maxleft,minright;
    float med;
    if(m>=n)
        {
            temp=n;
            n=m;
            m=temp;
            pt1=nums2;        //较长数组 
            pt2=nums1;        //较短数组 
        }
    else
    {
        pt1=nums1;
        pt2=nums2;
    }
    imin=0;
    imax=2*m;           //虚拟添加'#'分割
    
    while(imin<=imax)
    {
        i=imin+(imax-imin)/2;
        j=m+n-i;
        l1=(i==0)? INT_MIN : pt2[(i-1)/2];  //i==0,说明,短数组整体比长数组大,说明中位数在长数组中,此时,有可能i和j即为符合要求的分割,而此时l1不存在,因此给l1赋最小数,保证判断正确。下面赋最大值也是同样的考虑。 
        l2=(j==0)? INT_MIN : pt1[(j-1)/2];   //j==0,说明,长数组整体比短数组大,短数组整体比长数组小,i需要减小 ,j增加(i减小条件r2<l1)。 
        r1=(i==2*m)? INT_MAX : pt2[i/2];     //i==2*m,说明短数组整体比长数组小 , i需要减小(i减小条件r2<l1)。 
        r2=(j==2*n)? INT_MAX : pt1[j/2];     //j==2*n,说明短数组整体比长数组大, i需要增加 (增加条件l2>r1)。 
        
        if(l1<=r2&&l2<=r1)
            {
                maxleft=(l1>l2)? l1:l2;
                minright=(r1<r2)? r1:r2;
                return ((double)maxleft+(double)minright)/2;
            }
        else if(r2<l1)
            imax=i-1;
        else    
            imin=i+1;
    }
      return -1;
}

int main()
{
    int arr1[]={1,3,5,7,8,9};
    int arr2[]={10,11,12,13};
    int *p1,*p2;
    p1=arr1;
    p2=arr2;
    int med=0;
    med=findMedianSortedArrays(p1,6,p2,4);
    return 0;
    
}

参考文献
[1] https://leetcode.com/problems/median-of-two-sorted-arrays/#/description
[2] https://discuss.leetcode.com/topic/16797/very-concise-o-log-min-m-n-iterative-solution-with-detailed-explanation
[3]https://hk029.gitbooks.io/leetbook/%E4%BA%8C%E5%88%86%E6%9F%A5%E6%89%BE/035.%20Search%20Insert%20Position[M],md
[4]http://blog.csdn.net/hk2291976/article/details/51107778

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

推荐阅读更多精彩内容