给定两个大小分别为 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)