4. 寻找两个正序数组的中位数
题目描述
给定两个大小分别为 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
提示：
- nums1.length == m
- nums2.length == n
- 0 <= m <= 1000
- 0 <= n <= 1000
- 1 <= m + n <= 2000
- -106 <= nums1[i], nums2[i] <= 106
解题方法
- C 归并法
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) { int nums[nums1Size + nums2Size]; // 定义一个临时数组，用来存储合并后的数据 int cnt = 0; // 定义 cnt 用来做临时数组的指针 int i = 0, j = 0; // 定义 nums1、nums2 的数组指针 double result = 0.0; // 按顺序将数存放进临时数组内 while (i < nums1Size || j < nums2Size) { if (i == nums1Size) { nums[cnt++] = nums2[j++]; // nums1 处理完，添加 nums2 数据 } else if (j == nums2Size) { nums[cnt++] = nums1[i++]; // nums2 处理完，添加 nums1 数据 } else if (nums1[i] < nums2[j]) { nums[cnt++] = nums1[i]; // 按由小到大顺序数据 i++; } else { nums[cnt++] = nums2[j]; j++; } } if (cnt == 1) { result = nums[cnt - 1]; // 只有一个数，中位数 } else { if (cnt % 2 == 0) { // 偶数个数，中位数为中间两个数的平均值 result = (nums[cnt / 2] + nums[cnt / 2 - 1]) / 2.0; } else { // 奇数个数，中位数即为中间那个数 result = nums[cnt / 2]; } } return result; }
