给定两个大小为 m 和 n 的有序数组。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))
例:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5
分析:
引入第三个数组,将两个数组有序地放入第三个数组,找第三个数组的中间值,若第三个数组中的元素个数为偶数个,则取中间两个值的中间值,若为奇数,则就为中间的值。
代码实现:
public class median {
public double findMedianSortedArrays(int[] array1, int[] array2) {
int i=0,j=0,k=0;
int[] array3=new int[array1.length+array2.length]; //引入第三个数组
while(i<array1.length&&j<array2.length){
if(array1[i]<=array2[j]){
array3[k++]=array1[i++];
}
else if(array1[i]>=array2[j]){
array3[k++]=array2[j++];
}
}
while(i<array1.length){ //只有一个数组中有元素
array3[k++]=array1[i++];
}
while(j<array2.length){
array3[k++]=array2[j++];
}
//若数组中的元素为偶数个
if(array3.length%2==0){
double result=(array3[array3.length/2]+array3[(array3.length-1)/2])/2.0;
return result;
}
double result=array3[array3.length/2];
return result;
}
public static void main(String[] args) {
int[] array1={1,2,4,5,7,9};
int[] array2={1,4,5,7,8};
median m=new median();
System.out.println( "中间值为:" +m.findMedianSortedArrays(array1,array2));
}
}