首页 > 其他 > 详细

LeetCode Median of Two Sorted Arrays

时间:2015-10-17 07:01:54      阅读:225      评论:0      收藏:0      [点我收藏+]

原题链接在这里:https://leetcode.com/problems/median-of-two-sorted-arrays/

首先我们先明确什么是median,即中位数。 

引用Wikipedia对中位数的定义:

计算有限个数的数据的中位数的方法是:把所有的同类数据按照大小的顺序排列。如果数据的个数是奇数,则中间那个数据就是这群数据的中位数;如果数据的个数是偶数,则中间那2个数据的算术平均值就是这群数据的中位数。

所以要看 m+n是技术还是偶数,若是偶数,应该返回中间两个数的算术平方和。

在比较过程中,取nums1得中位数,nums2的中位数比较,若是nums1的中位数小,说明要找的中位数肯定不包括在 nums1的中位数和nums1中位数前面的数,为什么呢?如下是说明例子:

 将本体求中位数的要求调整为将两个数组整体合并排序后求其第k个数。下面用一个例子来说明:
技术分享
图一、例子示意图
      A班和B班的同学人数分别为aEnd与bEnd,有天体育老师说要找他们中间第k矮的同学,于是先将两班的同学按从小到大的顺序站成两队。假设k是为13。那此时A班派出第x矮的同学,B班派出第y矮的同学来比较(注意:此时x+y<=k)。
      若x为6,y为7。如果A的第6位同学比B的第7位同学还要矮,那如果两班整体排序,那A班的6位同学一定是要站在B班的第7位前面,最多A班第6位就站在B班第7位的前一位,最多就整体排第12,怎么都轮不到他排第k=13个。于是可以把A班前6位丢掉,在剩下的两个队列里面找。但是就不再是找第k个,而是找第k-x个了。因为x做了基数被丢掉了。
 
进入findKth后,为了方便比较若是前一个段长度比后一段下,就调换过来。当前面一个缩小到0时,就是取后面一个第k个即可。
前面取Math.min(k/2,m), k/2时因为k是int, 结果永远是向下取整,和m取小是因为 k/2大于前一段的长度时只保存m即可。当前一段取空了,后面一段多取一些。
递归过程中,会更改前一段长度,后一段长度和k, 所以终止条件有两个:一个是当较小一段长度 == 0时,取另一段的中点即可。另一个是当段缩减到1时,也就是 k == 1时,此时取两个元素中较小值就可以了。
比较时,若nums1中位数大,去掉nums1 前一段,并且在剩下部分中取第k-pa小值. 反之 nums2中位数大是做相对应的操作即可。多是两个数相同,那么就返回其中一个即可。

AC Java:

public class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int sum = m+n;
        if(sum%2 != 0){
            return findKth(nums1, 0, m-1, nums2, 0, n-1, sum/2 + 1);
        }else{
            return (findKth(nums1, 0, m-1, nums2, 0, n-1, sum/2) + findKth(nums1, 0, m-1, nums2, 0, n-1, sum/2 + 1))/2;
        }
    }
    private double findKth(int[] a, int aStart, int aEnd, int[] b, int bStart, int bEnd, int k){
        int m = aEnd - aStart + 1;
        int n = bEnd - bStart + 1;
        //统一长度,将短的array放到前面来
        if(m > n){
            return findKth(b,bStart,bEnd,a,aStart,aEnd,k);
        }
        //a is an empty array
        if(m == 0){
            return b[k-1];
        }
        //Stop condition, 都减到头了, 都剩下了一个元素
        if(k == 1){
            return Math.min(a[aStart],b[bStart]);
        }
        int pa = Math.min(k/2,m);
        int pb = k-pa;
        if(a[aStart+ pa -1] < b[bStart+pb-1]){
            return findKth(a,aStart+pa,aEnd,b,bStart,bEnd,k-pa);
        }else if(a[aStart+ pa -1] > b[bStart+pb-1]){
            return findKth(a,aStart,aEnd,b,bStart+pb,bEnd,k-pb);
        }else{
            return a[aStart+ pa -1];
        }
    }
}

 

参见这两篇帖子:http://www.cnblogs.com/springfor/p/3861890.html

http://blog.csdn.net/yutianzuijin/article/details/11499917

LeetCode Median of Two Sorted Arrays

原文:http://www.cnblogs.com/Dylan-Java-NYC/p/4886836.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!