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

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

猜你喜欢

转载自blog.csdn.net/qq_43069546/article/details/86488235
今日推荐