原题:
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
你可以假设 nums1 和 nums2 不同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
中位数是 (2 + 3)/2 = 2.5
解题思路一: 时间复杂度 O(k) 不满足题目的log(m+n),需要折半查找时间
private static double getMedianTwoSortedArr(int [] arr1,int [] arr2){
int index1 = 0, index2 = 0;
int[] arrAll = new int[arr1.length + arr2.length];
int k = 0;
double medianResult = Integer.MAX_VALUE;
int index_l = (arr1.length + arr2.length + 1) / 2;
while (k < index_l + 1 && index1 < arr1.length && index2 < arr2.length) {
if (arr1[index1] < arr2[index2]) {
arrAll[k] = arr1[index1++];
} else {
arrAll[k] = arr2[index2++];
}
if (k == index_l + 1) {
break;
}
k++;
}
while (k < index_l + 1 && index1 < arr1.length) {
arrAll[k] = arr1[index1++];
if (k == index_l + 1) {
break;
}
k++;
}
while (k < index_l + 1 && index2 < arr2.length) {
arrAll[k] = arr2[index2++];
if (k == index_l + 1) {
break;
}
k++;
}
medianResult = arrAll[index_l - 1];
System.out.println(arrAll[index_l - 1]);
if ((arr1.length + arr2.length) % 2 == 0) {
medianResult = (medianResult + arrAll[index_l]) *1.0 /2 ;
System.out.println(arrAll[index_l]);
}
return medianResult;
}