给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
简单思路:合并数组,并找到中位数即可,时间复杂度为O(m+n)
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] allNums = new int[nums1.length+ nums2.length];
int index = 0;
for(int i=0,j=0;i< nums1.length||j < nums2.length;){
if(i < nums1.length&& j < nums2.length){
if(nums1[i]<nums2[j]){
allNums[index++] = nums1[i++];
}else{
allNums[index++] = nums2[j++];
}
continue;
}
if(i < nums1.length){
allNums[index++] = nums1[i++];
}
if(j < nums2.length){
allNums[index++] = nums2[j++];
}
}
//System.out.println(Arrays.toString(allNums));
if(allNums.length%2==1){
return allNums[allNums.length/2];
}else{
return (allNums[allNums.length/2-1]+allNums[allNums.length/2])*1.0/2;
}
}
}
复杂思路: