LeetCode【4】. Median of Two Sorted Arrays --思路图解与java的不同方法实现

来源:互联网 发布:淘宝店怎么签约网红 编辑:程序博客网 时间:2024/05/01 19:45

Median of Two Sorted Arrays


       这道题确实有点难,想挺久,看别人答案也是不能一下子就明白。题目难度为Hard,原题如下:
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)).
      给定两个已排序的数组,求两个数组整体的中位数。时间复杂度要求为O(log (m+n))。

一、思路:

       将本体求中位数的要求调整为将两个数组整体合并排序后求其第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做了基数被丢掉了。
      当然,上述例子,只要(x+y)<=k的情况,x与y只要是为非负的,那都可以。只是为了效率,一般去x=y=(k/2)。而当某一队被丢到剩下人很少或者没人了,那这个取值就要根据具体情况来取值。

二、java 程序:

public class Solution {    public double findMedianSortedArrays(int[] nums1, int[] nums2) {        int m = nums1.length;        int n = nums2.length;        int k = m + n;                if((k&1)==1)        {            return find(nums1, 0, m, nums2, 0, n, k/2+1);        }else            return (find(nums1, 0, m, nums2, 0, n, k/2)+find(nums1, 0, m, nums2, 0, n, k/2+1))/2;    }    //递归算法,不断缩小两个数组的范围,同时k的值也相对两个搜索区间上限起始点而改变    public double find(int[] A, int aStart, int aEnd, int[] B, int bStart, int bEnd,int kth)    {    //1. 统一将长度短的放置于find函数参数的前面项        if(aEnd>bEnd)             return find(B, bStart, bEnd, A, aStart, aEnd, kth);            //2. 长度短的为空,则答案等同于求另外一个数组的中位数        if(aEnd<=0)             return B[bStart + kth -1];            //3. 递归到终点,两个数组的aStart和bStart已经到了中位数的位置        if(kth==1)            return min(A[aStart],B[bStart]);                int pa = min(kth/2,aEnd), pb = kth-pa;                if(A[aStart + pa-1]< B[bStart +pb -1])            return find(A, aStart+pa, aEnd-pa, B, bStart, bEnd, kth-pa );        else            return find(A, aStart, aEnd, B, bStart + pb, bEnd - pb,kth-pb);                    }    public int min(int a, int b)    {        return a>b?b:a;    }}




参考:LeetCode – Median of Two Sorted Arrays Java
                  Median of Two Sorted Arrays (JAVA)
1 0
原创粉丝点击