题目描述:
给定两个大小为 m 和 n 的有序数组 nums1
和 nums2
。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1
和 nums2
不会同时为空。
示例 1:
nums1 = [1, 3] nums2 = [2] 则中位数是 2.0
示例 2:
nums1 = [1, 2] nums2 = [3, 4] 则中位数是 (2 + 3)/2 = 2.5
直接上代码:
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) { int nums[5000], numsSize = nums1Size + nums2Size; int i = 0,j = 0,k = 0; //将两个有序的数组合并 for(i = 0; j < nums1Size && k < nums2Size; i++){ if(nums1[j] > nums2[k]) { nums[i] = nums2[k++]; } else { nums[i] = nums1[j++]; } } //将剩余的合并 while(j < nums1Size) nums[i++] = nums1[j++]; while(k < nums2Size) nums[i++] = nums2[k++]; //这里必须是2.0,因为返回的是double类型 if(numsSize % 2 == 0) { return (nums[numsSize / 2] + nums[numsSize / 2 - 1]) / 2.0; } else { return nums[numsSize / 2]; } }
注释已经给出完整解释!