给定两个大小分别为 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
思想:先按照归并排序进行合并两个有序表。然后按照合并后的顺序表偶数个和奇数个数据分别进行寻找中位数。
代码:
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
int *nums=(int*)malloc(sizeof(int)*(nums1Size+nums2Size));
int i=0,j=0,k=0;
while(i<=nums1Size-1&&j<=nums2Size-1){
if(nums1[i]<nums2[j]){
nums[k++]=nums1[i++];
}else{
nums[k++]=nums2[j++];
}
}
while(i<=nums1Size-1) nums[k++]=nums1[i++];
while(j<=nums2Size-1) nums[k++]=nums2[j++];
if((nums1Size+nums2Size)%2==0){
//数组下标从0开始
return ((double)nums[(nums1Size+nums2Size)/2]+(double)nums[(nums1Size+nums2Size)/2-1])/2;
}else{
return (double)nums[(nums1Size+nums2Size)/2];
}
}
时间复杂度O(log(m+n)),空间复杂度O(1)