[leetcode] Median of Two Sorted Arrays 两个排序数组找出第k个数,或者中位数

来源:互联网 发布:被淘宝大学骗了怎么办 编辑:程序博客网 时间:2024/05/18 03:56

先用图片讲下大概的原理:



下面是实现的代码:

代码基本上是按照leetcode中的某位大神的c++代码改写为java的

地址为:https://leetcode.com/discuss/30807/o-lg-m-n-c-solution-using-kth-smallest-number

public class MedianOfTwoSortedArrays{    public int kth(int a[],int sa,int lena, int b[],int sb,int lenb,int k){        if(lena < lenb) return kth(b,sb,lenb,a,sa,lena,k); //将较短的数组放置到后面        //上一行的代码的理由是,保证 int j = Math.min(lenb,k/2) 是对小数组进行处理        //如果比进行调换,int j = Math.min(lenb,k/2) 对大的数组进行处理,得出j=k/2 i=k/2 将可能造成数组的越界        if(lenb == 0) return a[sa+k-1]; //其中一个数组为0 那么k值必定在另一个数组中        if(k == 1) return Math.min(a[sa],b[sb]); //如果k=1,那么就其中的最小者                int j = Math.min(lenb,k/2);//保证不会越界        int i = k-j;        if(a[sa+i-1]>b[sb+j-1]) return kth(a,sa,i,b,sb+j,lenb-j,k-j);        else return kth(a,sa+i,lena-i,b,sb,j,k-i);    }    public double findMedianSortedArrays(int[] nums1, int[] nums2) {        int len1 = nums1.length;          int len2 = nums2.length;        int k = (len1+len2)/2;        int result1 = kth(nums1,0,len1,nums2,0,len2,k+1);        if((len1+len2)%2 == 0){             int result2 = kth(nums1,0,len1,nums2,0,len2,k);            return((result1+result2)*1.0/2); //这里求的是两个数的均值        }        return result1;    }}


0 0
原创粉丝点击