Median of Two Sorted Arrays(获取两个有序数列的中值)

原创 2017年09月11日 19:29:26

转载注明出处:http://blog.csdn.net/xiaohanluo/article/details/77936830

Median of Two Sorted Arrays

最近没事刷LeetCode,有一题很有意思,取中值问题。

中值是指将统计总体当中的各个变量值按大小顺序排列起来,形成一个数列,处于变量数列中间位置的变量值就称为中位数。当数列的项数N为奇数时,处于中间位置的变量值即为中位数;当数列的项数N为偶数时,中位数则为处于中间位置的2个变量值的平均数。

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

题目意思是给定两个排好序的数组,然后找到这两个数组的中值,要求时间复杂度是O(log (m+n))。

正常思路是将两个数组合并成一个新的有序数组,然后直接找到中值就行了,时间复杂度是O(m+n),但是题目要求时间复杂度是O(log (m+n)),这就比较有意思了。

思路

正常解决是先排序再求中值,但是由于只需要拿到中值,所以理论上只需要排一半的序就可以获取中值,但是排序的话时间复杂度达不到O(log (m+n))的要求。

再深入理解一下,获取中值,可以理解成拿到数组中的第k个数。数组元素数为奇数,中值就是length/2 + 1的位置的数;当为元素数为偶数,中值就是length/2和length/2 + 1两个位置的平均值。

理清思路之后就可以开始着手解决了。

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    int totalLength = nums1.length + nums2.length;

    if((totalLength & 0x1) == 1) { // 奇数
        return findMedian(nums1, nums1.length, nums2, nums2.length, totalLength/2 + 1);
    } else {
        return (findMedian(nums1, nums1.length, nums2, nums2.length, totalLength/2)
                + findMedian(nums1, nums1.length, nums2, nums2.length, totalLength/2 + 1)) /2d;
    }
}

解题思路取了巧,看到O(log (m+n))复杂度的时候,立马想到了二分法。

private static double findMedian(int[] longerArray, int m, int[] shorterArray, int n, int k) {
    if(n > m) { // 保证第一个数组是比较长的数组
        return findMedian(shorterArray, n, longerArray, m, k);
    }

    if(n == 0) { // 当短数组为空时候,直接从长数组中获取数值
        return longerArray[k - 1];
    }

    if(k == 1) { // 当数值位置在第一位时,比较两个数组获取最靠前的数值
        return Math.min(longerArray[0], shorterArray[0]);
    }

    // 根据比例二分那个数组
    int mk = (int)(m*k/(m + n)*1f);
    int nk = k - mk;

    if(longerArray[mk - 1] > shorterArray[nk - 1]) {
        int[] newLonger = Arrays.copyOfRange(longerArray, 0, mk);
        int[] newShort = Arrays.copyOfRange(shorterArray, nk, n);
        //去除了比较小的一部分,相当于整个数值向左移动了nk位
        return findMedian(newLonger, newLonger.length, newShort, newShort.length, k - nk);
    } else if(longerArray[mk - 1] < shorterArray[nk - 1]) {
        int[] newLonger = Arrays.copyOfRange(longerArray, mk, m);
        int[] newShort = Arrays.copyOfRange(shorterArray, 0, nk);
        //去除了比较小的一部分,相当于整个数值向左移动了nk位
        return findMedian(newLonger, newLonger.length, newShort, newShort.length, k - mk);
    } else {
        return longerArray[mk - 1];
    }
}

最后附上官方解决方法,没有使用递归。https://leetcode.com/articles/median-of-two-sorted-arrays/

附上其他解决此题的方法,部分算法不符合O(log (m+n))要求。http://blog.csdn.net/whucyl/article/details/23524045

版权声明:本文为博主原创文章,未经博主允许不得转载。

ubuntu14.04安装opencv 3.1.0

sudo apt-get install cmake cmake --version sudo apt-get update sudo apt-get install build-essent...
  • jjff46
  • jjff46
  • 2016年01月07日 14:28
  • 4317

超级星计划

超级星计划: 1.向大神学习、向身边的人学习,但首先得有自己的学习计划 2.专栏《一个程序员的进化史》给了我很多鼓励,但我必须首先有所成长,才能实时分享自己的进化史。 3.良好的品质不好养成。也...

LeetCode 4. Median of Two Sorted Arrays(两个有序数组的中位数)

原题网址:https://leetcode.com/problems/median-of-two-sorted-arrays/ There are two sorted arrays nums1 an...
  • jmspan
  • jmspan
  • 2016年05月01日 15:39
  • 550

[Leetcode #4]Median of Two Sorted Arrays 计算两个有序数组的中位数

原题地址:https://leetcode.com/problems/median-of-two-sorted-arrays/ 题目要求是:给定两个有序数组nums1[m]和nums2[n],计算它...

两个有序数组的中位数Median of Two Sorted Arrays(很重要)

https://leetcode.com/problems/median-of-two-sorted-arrays/ 对于一个长度为n的已排序数列a,若n为奇数,中位数为a[n / 2 + 1]...

Leetcode4: Median of Two Sorted Arrays (求已排序两个数组的中间值)

最开始的想法是利用归并排序的归并merge函数将数组先完全排序,然后根据奇偶个数来决定中间值的大小,思路清晰,代码简单,如下:public class Solution { public do...

寻找两个不等长数组的中位数 Median of Two Sorted Arrays

题目源自于Leetcode。经典好题。 题目: 给出两个有序数组,长度不一定相同,一个是m一个是n,要求给出他们合并在一起之后的数组的中位数。 要求时间复杂度为O(log(m+n)),所以不可以合并数...

LeetCode | Median of Two Sorted Arrays(两个数组的中位数)

题目: There are two sorted arrays A and B of size m and n respectively. Find the median of the two so...

LeetCode Median of Two Sorted Arrays 在两个已排列的数组中找出中位数。时间复杂度为O(log(min(N,M))

完成一道题,真的是好开心,虽然自己蠢蠢的,到了研究生还在弄这些基础的算法,才开始学去整理自己的知识。但是没关系,就算是要用几年的时间才能获得别人现在就拥有的知识能力,能就用几年的时间吧!题目:Ther...

【算法导论学习-016】两个已排过序的等长数组的中位数(median of two sorted arrays)

求两个已排过序的等长的数组的中位数(median of two sorted arrays)
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:Median of Two Sorted Arrays(获取两个有序数列的中值)
举报原因:
原因补充:

(最多只允许输入30个字)