LeetCode之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) {
multiset<int> s_array;
double ret = 0.0;
s_array.insert(nums1.begin(), nums1.end());
s_array.insert(nums2.begin(), nums2.end());
vector<int> nums3(s_array.begin(), s_array.end());
int size_count = s_array.size();
if (size_count % 2 == 0)
{
ret = (nums3.at(size_count / 2) + nums3.at(size_count / 2 - 1)) / 2.0;
}
else
{
ret = nums3.at((size_count - 1) / 2);
}
return ret;
}
};