先用图片讲下大概的原理:
下面是实现的代码:
代码基本上是按照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;
}
}