题目
给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。
进阶:你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗?
示例 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
此题我使用的是归并算法,归并的话时间和空间复杂度都是o(m+n),首先通过判断两个数组内绳子的大小,一个个的插入新数组,就得到了一个非递减的数组,然后通过判断这个数组里面数字的个数是奇数还是偶数,返回中位数。
很简单,所以直接贴代码。
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int[] nums3=new int[nums1.length+nums2.length];
int i=0,j=0,k=0;
while(i<nums1.length&&j<nums2.length){
if(nums1[i]<=nums2[j]){
nums3[k++]=nums1[i++];
}else{
nums3[k++]=nums2[j++];
}
}
while(i<nums1.length){
nums3[k++]=nums1[i++];
}
while(j<nums2.length){
nums3[k++]=nums2[j++];
}
double result=0;
if(nums3.length>1){
result =nums3.length%2==1?nums3[nums3.length/2]:
((double)(nums3[nums3.length/2]+nums3[nums3.length/2-1])/2);
}else{
result=nums3[0];
}
return result;
}
}