题目要求:给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
示例:nums1 = [1, 2];nums2 = [3, 4];则中位数是 (2 + 3)/2 = 2.5
方法一:合并法,虽然通过,但是时间复杂度不符合要求
思路:参考合并两个链表 时间复杂度O(max(m,n)),空间复杂度O(m+n)
【1】用两个指针分别指向数组的头节点。将两个数组合并成一个数组,时间复杂度O(max(m,n)),空间复杂度O(m+n)
【2】对合并完的有序数组进行取中位数
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
vector<int> vec;
//iter1指向数组nums1,iter2指向数组nums2。对两数组进行遍历
vector<int>::iterator iter1 = nums1.begin(), iter2 = nums2.begin();
while (iter1!=nums1.end()&&iter2!=nums2.end()) {
if (*iter1 < *iter2) {
vec.push_back(*iter1);
++iter1;
}
else {
vec.push_back(*iter2);
++iter2;
}
}
//nums1未遍历完,则nums2一定遍历完了
if (iter1 != nums1.end()) {
//将未遍历完的数添加至vec
while (iter1 != nums1.end()) {
vec.push_back(*iter1);
++iter1;
}
}
else {
while (iter2 != nums2.end()) {
vec.push_back(*iter2);
++iter2;
}
}
//若数组长度为偶数,则需进行求平均值
if (vec.size() % 2 == 0) {
int mid = vec.size() / 2;
return ((double)vec.at(mid) + vec.at(mid - 1)) / 2;
}
else {
int mid = (vec.size() - 1) / 2;
return (double)vec.at(mid);
}
}
方法二:递归法
该方法在leetcode官方题解中给出,图文并茂,容易理解。
链接:https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/