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

最新推荐文章于 2025-06-26 14:29:51 发布

原创最新推荐文章于 2025-06-26 14:29:51 发布 · 253 阅读

0 ·

CC 4.0 BY-SA版权

Tzt的打工时间

文章标签：

#leetcode #算法 #职场和发展

HOT-100 专栏收录该内容

33 篇文章

订阅专栏

这篇博客讨论了一种算法，用于在O(log(m+n))的时间复杂度内找到两个正序数组的中位数。示例展示了如何将两个数组合并并计算中位数，包括当数组长度为奇数和偶数时的情况。该算法适用于数据处理和排序问题。

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

寻找两个正序数组的中位数
给定两个大小分别为 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

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size(), n = nums2.size();
        vector<int> merge(m + n);
        int left_1 = 0, left_2 = 0, pos = 0;
        while(left_1 < m && left_2 < n){
            // nums1[left_1] > nums2[left_2] ? merge[pos++] : merge[];
            merge[pos++] = nums1[left_1] < nums2[left_2] ? nums1[left_1++] : nums2[left_2++];
        }
        while(left_1 < m){
            merge[pos++] = nums1[left_1++];
        }
        while(left_2 < n){
            merge[pos++] = nums2[left_2++];
        }
        return (m + n) % 2 == 1 ? merge[(m + n) / 2] : (merge[(m + n) / 2] + merge[(m + n) / 2 - 1]) / 2.0;
    }
};