题目
给定两个大小为 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
代码
- 直接组成一个新的有序数组,时间复杂度:
O(n1 + n2)
#include<vector>
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
double ans = 0.0;
vector<int> res;
auto t1 = nums1.begin();
vector<int>::iterator t2 = nums2.begin();
while (t1 != nums1.end() && t2 != nums2.end())
{
if (*t1 <= *t2)
{
res.push_back(*t1);
t1++;
}
else
{
res.push_back(*t2);
t2++;
}
}
if (t1 != nums1.end())
{
res.insert(res.end(), t1, nums1.end());
}
if (t2 != nums2.end())
{
res.insert(res.end(), t2, nums2.end());
}
/*while (t1 != nums1.end())
{
res.push_back(*t1);
t1++;
}*/
/*while (t2 != nums2.end())
{
res.push_back(*t2);
t2++;
}*/
if (res.size() % 2 == 0)
{
//偶数
int a = res.size() / 2;
ans = (res[a] + res[a - 1]) / 2.0;
}
else
{
//奇数
int a = res.size() / 2;
ans = res[a];
}
return ans;
}
};
- 中分法,时间复杂度为:
O(log(m + n))
待更新。。。