给定两个大小为 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)
这里我看到这个问题最先想到的就是将这两个数组合并然后对合并成的新数组进行排序,然后再从新的数组中寻找中位数:(这样柑橘问题不是很简单吗?)
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
vector<int> v;
double num = 0.0;
double n = 2.0;
for (int i = 0; i < nums1.size(); i++)
{
v.push_back(nums1[i]);
}
for (int i = 0; i < nums2.size(); i++)
{
v.push_back(nums2[i]);
}
sort(v.begin(), v.end());
if ((nums1.size() + nums2.size()) % 2 == 0)
{
int mid = (nums1.size() + nums2.size()) / 2.0;
num = (v[mid] + v[mid - 1]) / n;
}
else
{
int mid = (nums1.size() + nums2.size()) / 2;
num = v[mid];
}
return num;
}
};
但是这道题在LeedCode上显示的难度为困难,我仔细一看题目还有复杂度的要求,因为在这里我使用的是sort排序采用的是快排的算法,复杂度为(m+n)log(m+n)所以很显然我虽然通过了但是我的复杂度并不满足,所以我将sort的排序去掉了,在合并成到一个容器是就已经将其排好序
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
vector<int> v;
double num = 0.0;
double n = 2.0;
int i = 0, j = 0;
while (i < nums1.size() && j < nums2.size())
{
if (nums1[i] < nums2[j])
{
v.push_back(nums1[i]);
i++;
}
else
{
v.push_back(nums2[j]);
j++;
}
}
while (i < nums1.size())
{
v.push_back(nums1[i]);
i++;
}
while (j < nums2.size())
{
v.push_back(nums2[j]);
j++;
}
if ((nums1.size() + nums2.size()) % 2 == 0)
{
int mid = (nums1.size() + nums2.size()) / 2.0;
num = (v[mid] + v[mid - 1]) / n;
}
else
{
int mid = (nums1.size() + nums2.size()) / 2;
num = v[mid];
}
return num;
}
};
按照这种方式其算法复杂度降低到了O(N)但是还是没有达到Olog(m+n)所以我现在就彻底的明白了这道题的为甚什么等级为困难了。由于能力有限就先到这里,后文待续…
具体请参考力扣里面的题解。