LeetCode 4 寻找两个有序数组的中位数（C数组拼接）

最新推荐文章于 2024-07-08 22:58:13 发布

huihui_8

最新推荐文章于 2024-07-08 22:58:13 发布

阅读量161

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/huihui_8/article/details/104301922

本文探讨了如何在两个已排序的数组中找到中位数，提出了一个简单易懂的解决方案，通过合并并排序两个数组，然后根据数组长度的奇偶性确定中位数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.题目链接

https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

2.题目描述

给定两个大小为 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

3.题目解析
1）中位数的含义

将一个集合划分为两个长度相等的子集，其中一个子集中的元素总是大于另一个子集中的元素。

2）题解

根据对中位数的理解，可以将2个有序数组拼接为新数组，重新排序，新数组长度为偶数时，中位数即为：

media = nums[k / 2]；

新数组长度为奇数时，中位数即为：

media = (nums[k / 2 - 1] + nums[k / 2]) / 2;

4.代码实现

int comp(const void *a, const void *b){
    return *(int*)a - *(int*)b;
}

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size){
    double media;
    int *nums = (int*)malloc((nums1Size + nums2Size) * sizeof(int));
    for (int i = 0; i < nums1Size; i++) {
        nums[i] = nums1[i];
    }
    for (int i = 0; i < nums2Size; i++) {
        nums[nums1Size + i] = nums2[i];
    }
    qsort(nums, nums1Size + nums2Size, sizeof(int), comp);
    int k = nums1Size + nums2Size;
    if (k % 2 == 0)
        media = (float)(nums[k / 2 - 1] + nums[k / 2]) / 2;
    else
        media = nums[k / 2]	;
    return media;
}