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

最新推荐文章于 2024-08-27 09:15:46 发布

原创最新推荐文章于 2024-08-27 09:15:46 发布 · 114 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

75 篇文章

订阅专栏

归并

6 篇文章

订阅专栏

本文介绍了一种求解两个有序数组中位数的算法，通过将两个数组合并成一个有序数组，然后根据数组长度的奇偶性找到中位数。示例展示了当输入数组长度不同时，如何正确计算中位数。

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

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

/*采用归并排序的方法，合并两个有序数组再取中位数*/
class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2)
    {
        vector <double> nums3;
        int size1=nums1.size();
        int size2=nums2.size();
        if(size1==0)
        {
            if(size2%2==1)return  nums2.at(size2/2);
            else return (nums2.at(size2/2)+nums2.at(size2/2-1))/2.0;
        }
        if(size2==0)
        {
            if(size1%2==1)return  nums1.at(size1/2);
            else return (nums1.at(size1/2)+nums1.at(size1/2-1))/2.0;
        }
        while(size1!=0&&size2!=0)
        {
            if(nums1.at(size1-1)>nums2.at(size2-1))
            {
                nums3.push_back(nums1.at(size1-1));
                size1--;
            }
            else
            { 
                nums3.push_back(nums2.at(size2-1));
                size2--;
            }
        }
       while(size1!=0)
        {
            nums3.push_back(nums1.at(size1-1));
            size1--;
        }
        while(size2!=0)
        {
            nums3.push_back(nums2.at(size2-1));
            size2--;
        }
        int size3=nums3.size();
        if(size3%2==1)
            return nums3.at(size3/2);
        else
            return (nums3.at(size3/2)+nums3.at(size3/2-1))/2.0;
    }
};