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

标签: leetcode中值有序数组两个Median
126人阅读 评论(0) 收藏 举报
分类:

转载注明出处: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

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:107360次
    • 积分:1391
    • 等级:
    • 排名:千里之外
    • 原创:29篇
    • 转载:0篇
    • 译文:1篇
    • 评论:125条
    最新评论