题目描述
给定两个大小为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
解题思路
题目已经明确说明,两个字数组不会同时为空。因此我们需要考虑的情况有其中一个数组为空和两个数组都不为空两种情况。当其中一个数组为空时,返回另一个数组的中位数即可。如果两个数组都不为空,我们就对两个数组进行排序再找出中位数即可。
Java程序
public class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
//如果nums1数组为空,直接返回nums2的中位数
if(nums1 == null || nums1.length == 0){
//如果nums2的元素个数为奇数个,直接返回中间元素
if(nums2.length % 2 == 1)
return nums2[nums2.length / 2];
else {//如果nums2的元素个数为偶数个,直接返回中间两个的平均值
double median = ((double)(nums2[nums2.length / 2] + nums2[(nums2.length / 2) -1]))/2;
return median;
}
//如果nums2为空,直接返回nums1的中位数
}else if(nums2 == null || nums2.length == 0){
if(nums1.length % 2 == 1)
return nums1[nums1.length / 2];
else {
double median = ((double)(nums1[nums1.length / 2] + nums1[(nums1.length / 2) -1]))/2;
return median;
}
}else{
ArrayList<Integer> array = new ArrayList<>();
int j = 0;
for(int i = 0;i <= nums1.length;i++) {
//如果nums2有剩余,将剩余元素直接放在集合末尾
if (i == nums1.length && j != nums2.length) {
for(int m = j;m < nums2.length;m++)
array.add(nums2[m]);
} //如果nums1有剩余,将剩余元素直接放在集合末尾
else if(j == nums2.length && i != nums1.length){
for(int m = i;m < nums1.length;m++)
array.add(nums1[m]);
break;
}//将nums1和nums2中元素按照从小到大顺序放在集合中
else if(i != nums1.length){
if (nums1[i] <= nums2[j]) {
array.add(nums1[i]);
} else {
array.add(nums2[j]);
i = i-1;
j++;
}
}
else break;
}
if(array.size() % 2 == 1)
return array.get(array.size()/2);
else {
double meidan = ((double)(array.get(array.size() / 2)+array.get(array.size() / 2 -1)))/2;
return meidan;
}
}
}
解题要点
POINT 1:如何判断数组为空?
例如:int[] num = {};
判断该数组为空的条件为num == NULL 或者 num.length == 0.
POINT 2:求数组中位数时应该分两种情况
情况1:数组中的元素个数为奇数。由于所给数组为有序数组,当数组元素个数为奇数时直接返回最中间一个元素值即可,即数组下标为num.length/2的元素
情况2:当数组种元素个数为偶数时。返回中间两个元素的平均值即可。即返回下标为num.length/2-1和num.length/2两个元素的平均值。
POINT 3:int型数据做分子时需要考虑是否需要将int型数据转换为double型数据
例如:double num = 3/2;//num的值为1
int num = 3/2;//num的值为1
int num = 3.0/2;//num的值为1
要想让一个int型数据除以2为double类型,必须将该int型数据强制转换为double型数据。即 double num = ((double)3)/2或double num = 3.0/2.
POINT 4:如何判断数组有剩余元素?
在将连个数组元素排序存入ArrayList集合中时,总会存在一个数组已经全部存入而另一个数组还有剩余的情况。此时,就需要我们进行判断到底是那个数组有剩余,并将有剩余的数组的剩余元素全部存入集合末尾。
我们以本题为例,如果nums1有剩余则j == nums2.length && i != nums1.length(本题中我们用j作为nums2数组的索引,用i作为nums1数组的索引)。这也就是为什么我们要将for循环中i的值取到nums1.length的原因,最后再用break跳出最后一次循环。
以上就是关于如何求两个有序数组的Java程序,大家如果有什么疑问或者建议可以直接评论。