4. 寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
示例 3:
输入:nums1 = [0,0], nums2 = [0,0]
输出:0.00000
示例 4:
输入:nums1 = [], nums2 = [1]
输出:1.00000
示例 5:
输入:nums1 = [2], nums2 = []
输出:2.00000
思路:用归并排序先排序,再求中位数
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] nums=mergeSort(nums1,nums2);
int size=nums.length;
if(size%2!=0){
return nums[size/2];
}else{
int mid=size/2;
return (nums[mid-1]+nums[mid])/2.0;
}
}
public int[] mergeSort(int[] nums1,int[] nums2){
int[] nums=new int[nums1.length+nums2.length];
System.out.println(nums.length);
int L1=0;
int L2=0;
int i=0;
while(L1<nums1.length&&L2<nums2.length){
if(nums1[L1]<=nums2[L2]){
nums[i++]=nums1[L1++];
}else{
nums[i++]=nums2[L2++];
}
}
while(L1<nums1.length){
nums[i++]=nums1[L1++];
}
while(L2<nums2.length){
nums[i++]=nums2[L2++];
}
return nums;
}
}

浙公网安备 33010602011771号