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