4. 寻找两个正序数组的中位数

题目描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

示例

示例 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

来源：力扣（LeetCode）
链接：leetcode-cn.com/problems/me…

实现

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
    int v = 0;    // 标记新数组的下标
    int nums3[nums1Size + nums2Size];
    int result = 0;
    int i = 0;
    int j = 0;
    if (nums1 == NULL && nums2 == NULL) {
        return 0.0;
    }
    
    for (i = 0, j = 0; i < nums1Size && j < nums2Size;) {
        if (nums1[i] >= nums2[j]) {
            nums3[v++] = nums2[j++]; 
        } else {
            nums3[v++] = nums1[i++];
        }
    }
    if (i < nums1Size) {
        while (i < nums1Size) {
            nums3[v++] = nums1[i++];
        }
    }
    if (j < nums2Size) {
        while (j < nums2Size) {
            nums3[v++] = nums2[j++];
        }
    }
    if ((nums1Size + nums2Size) % 2 == 0) {
        return ((nums3[(nums1Size + nums2Size) / 2 - 1]) + nums3[(nums1Size + nums2Size) / 2]) / 2.0;
    } else {
        return nums3[(nums1Size + nums2Size) / 2];
    }
    return -1.0;
}
复制代码