LeetCode 4

Median of Two Sorted Arrays

题目:

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)).
You may assume nums1 and nums2 cannot be both empty.

描述:两个已经排序好的数组,数组不是空,求中位数。要求算法的时间复杂度是O(long(m+n))。

分析:

1、如果不看时间复杂度的话,可以使用暴力方法实现:把两个数组拼接,然后重新排序,寻找中位数,较为简单,使用Java实现。
2、要是考虑复杂度的话,较为困难,后面会用C语言具体分析。

1、Java实现

内容简单,这里不做解释。

public double findMedianSortedArrays(int[] nums1, int[] nums2) {

        double median = 0.0f;
        int m = nums1.length;
        int n = nums2.length;
        int[] newNums = Arrays.copyOf(nums1, nums1.length + nums2.length);

        System.arraycopy(nums2, 0, newNums, nums1.length, nums2.length);
        Arrays.sort(newNums);

        if ((m + n) % 2 == 0) {
            median = (newNums[(m + n) / 2] + newNums[(m + n) / 2 - 1]) / 2.0;
        } else {
            median = newNums[(m + n) / 2];
        }

        return median;
    }

运行时间:


image.png

时间复杂度:主要是Arrays.sort()使用的快排,时间复杂度为O( (n+m)*log(n+m) )

空间复杂度:O(n+m)

2、 C语言实现(1)

新建一个长度为 (m+n)/2 +1 的数组newnumsi,j为先指向数组nums1mums2的第0个元素,通过i,j的移动把两个源数组的按照升序复制到新数组newnums中。

  • 1. 处理源数组是空的情况,一个如果是空,另一个直接复制数组到newnums中。

  • 2. 两个都不为空,开始循环,循环次数为length/2+1(总长度一半+1)

  • 3. 控制移动边界i>=num1Size时,nums1到最后一个元素,从nums2赋值,j>=num2Size时,nums2到最后一个元素,从nums1赋值。
    判断nums1[i] < nums2[j],每次都把较小的元素赋值到新数组中。

  • 4.分偶数和奇数判断中位数情况。

double findMedianSortedArrays(int *nums1, int nums1Size, int *nums2, int nums2Size) {

    int length = nums1Size + nums2Size;
    int *newnums = (int *) malloc(sizeof(int) * (length / 2 + 1));
    int index = 0, i = 0, j = 0;


    double median;
    if (nums1Size == 0) {
        newnums = nums2;
    } else if (nums2Size == 0) {
        newnums = nums1;
    } else {

        while (index <= length / 2) {

            if (i >= nums1Size) {
                newnums[index] = nums2[j];
                j++;
            } else if (j >= nums2Size) {
                newnums[index] = nums1[i];
                i++;
            } else {
                if (nums1[i] < nums2[j]) {
                    newnums[index] = nums1[i];
                    i++;
                } else {
                    newnums[index] = nums2[j];
                    j++;
                }
            }

            index++;
        }
    }

    if (length % 2 == 0) {
        median = (newnums[length / 2 - 1] + newnums[length / 2]) / 2.0;
    } else {
        median = newnums[length / 2];
    }
    return median;
}

运行时间:


image.png

时间复杂度:O((n+m)/2+1) = O(n)

空间复杂度:O((n+m)/2+1) = O(n)

3、C语言实现(2)

中位数的定义是把一个集合分为左右长度相等的两个子集合,所以我们如果保证两个数组的,左半部分的长度=右半边分长度,就可以找出中位数。
这时候要保证:

len(left_part) = len(right_part)
max(left_part) <=min(right_part)

len(left_part) = i
len(right_part)) = j
i = m - i
j = n - j

因此我们需要确保

  • i + j = m - i +n - j (or i + j = m - i + n - j + 1)
    if m<=n 我们需要设置 i ~(0,m) , j = (m + n + 1)/2 - i
  • nums2[j−1]≤nums1[i] 和 nums1[i−1]≤nums2[j]

我们需要先确保m<=n,如果m>n时,j可能会小于 0,然后使用二分法查询数组,处理好四种边界问 题i = 0 ,j = 0 ,i = m 或 j = n的情况,具体细节看代码。

double findMedianSortedArrays(int *nums1, int nums1Size, int *nums2, int nums2Size) {

    int m = nums1Size;
    int n = nums2Size;

    if (m > n) { // to ensure m<=n
        int *temp = nums1;
        nums1 = nums2;
        nums2 = temp;

        int tmp = m;
        m = n;
        n = tmp;
    }

    int iMin = 0, iMax = m, halfLen = (m + n + 1) / 2;

    while (iMin <= iMax) {
        int i = (iMin + iMax) / 2;
        int j = halfLen - i;

        if (i < iMax && nums2[j - 1] > nums1[i]) {
            iMin = i + 1; // i is too small
        } else if (i > iMin && nums1[i - 1] > nums2[j]) {

            iMax = i - 1; // i is too big

        } else { // i is perfect

            int maxLeft = 0;

            if (i == 0) {
                maxLeft = nums2[j - 1];
            } else if (j == 0) {
                maxLeft = nums1[i - 1];
            } else {
                maxLeft = nums1[i - 1] > nums2[j - 1] ? nums1[i - 1] : nums2[j - 1];
            }

            if ((m + n) % 2 == 1) {
                return maxLeft;
            }

            int minRight = 0;

            if (i == m) {
                minRight = nums2[j];
            } else if (j == n) {
                minRight = nums1[i];
            } else {
                minRight = nums2[j] > nums1[i] ? nums1[i] : nums2[j];
            }

            return (maxLeft + minRight) / 2.0;
        }
    }
    return 0.0;
}

耗时如图:


image.png

还是比较不错的。
时间复杂度:O(log(min(m,n))).
空间复杂度:O(1)

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

推荐阅读更多精彩内容

  • <center>#1 Two Sum</center> link Description:Given an arr...
    铛铛铛clark阅读 2,082评论 0 3
  • There are two sorted arrays nums1 and nums2 of size m and...
    lemooon阅读 322评论 0 0
  • 今天刚下载简书的App,觉得总需要有个媒介纪录一下生活的痕迹,好像这个挺适合的,和老公从大学恋爱到现在五年了,直到...
    芬外饶阅读 182评论 0 2
  • 约定二:与身边人和约定 你我等约定就是我们俩共同的主 1、大家共同的约定形成社会公德 如:大家共同遵守幸福百合核心...
    冯岚阅读 131评论 0 0
  • 萧风冷瑟心不快,日月芒里单影在。 花丛群蝶无一属,绿柳园中不具才。
    路里89阅读 206评论 0 3