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;
}
复制代码
分类:
后端
标签: