给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。
请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。
示例 1:
nums1 = [1, 3] nums2 = [2] 中位数是 2.0
示例 2:
nums1 = [1, 2] nums2 = [3, 4] 中位数是 (2 + 3)/2 = 2.5
正确代码:A4两个排序数组的中位数.java
package LeetCode;
import java.util.ArrayList;
public class A4两个排序数组的中位数 {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n1 = nums1.length;
int n2 = nums2.length;
int i = 0;
int j = 0;
int k = 0;
int[] nums = new int[n1 + n2];
while (n1 != 0 && n2 != 0) { // 公共长度先对比,小的放在新的数组里
nums[k] = (nums1[i] < nums2[j]) ? nums1[i] : nums2[j];
if (nums1[i] < nums2[j]) {
i++;
k++;
n1--;
} else {
j++;
k++;
n2--;
}
}
if (n1 == 0) { // 如果nums1比较小,把nums2剩下的放入nums中
while ((n2--) != 0) {
nums[k] = nums2[j];
j++;
k++;
}
}else if (n2 == 0) { // 如果nums2比较小,把nums1剩下的放入nums中
while ((n1--) != 0) {
nums[k] = nums1[i];
i++;
k++;
}
}
if ((nums1.length + nums2.length) % 2 == 0) { // 如果数字个数为偶数
return (nums[(nums1.length + nums2.length) / 2] + nums[(nums1.length + nums2.length) / 2 - 1]) /2.0;
} else { // 如果数字个数为奇数
return nums[(nums1.length + nums2.length) / 2];
}
}
public static void main(String[] args) {
int[] nums1 = {};
int[] nums2 = { 2, 3 };
A4两个排序数组的中位数 a4两个排序数组的中位数 = new A4两个排序数组的中位数();
System.out.println(a4两个排序数组的中位数.findMedianSortedArrays(nums1, nums2));
}
}
需要注意的地方如下所示:
if (n1 == 0) { // 如果nums1比较小,把nums2剩下的放入nums中
while ((n2--) != 0) {
nums[k] = nums2[j];
j++;
k++;
}
} else if (n2 == 0) { // 如果nums2比较小,把nums1剩下的放入nums中
while ((n1--) != 0) {
nums[k] = nums1[i];
i++;
k++;
}
}
*如果不加else,上面那个判断n2--后,n2变成0,又会满足下面那个判断的条件。
return (nums[(nums1.length + nums2.length) / 2] + nums[(nums1.length + nums2.length) / 2 - 1]) / 2.0;
*减1是因为下标从0开始。
*除以2.0是为了让结果不要取整。