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

示例 3:

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例 4:

输入：nums1 = [], nums2 = [1]
输出：1.00000

示例 5:

输入：nums1 = [2], nums2 = []
输出：2.00000

三、输入输出说明

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

四、基本思路

同时遍历两个数组，每次将两者更小的元素放入合并数组中，然后分奇偶情况考虑合并数组的中位数，时间复杂度为O(log (m+n))

五、代码

C语言：

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    int nums3[2005];
    int index = 0;//合并数组中的索引
    int i, j;//分别遍历nums1和nums2
    double mid;
    for (i = 0, j = 0; i < nums1Size && j < nums2Size; ) {
        if (nums1[i] < nums2[j]) {
            nums3[index++] = nums1[i++];
        }
        else {
            nums3[index++] = nums2[j++];
        }
    }
    while (i < nums1Size) {
        nums3[index++] = nums1[i++];
    }
    while (j < nums2Size) {
        nums3[index++] = nums2[j++];
    }
    if (index%2 == 1) {
        //总个数为奇数
        mid = nums3[index/2];
    }
    else {
        mid = (nums3[index/2 - 1] + nums3[index/2]) / 2.0;
    }
    return mid;
}