1、题目:寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
示例 1:
输入:nums1 = [1,3], nums2 = [2]
输出:2.00000
解释:合并数组 = [1,2,3] ,中位数 2
示例 2:
输入:nums1 = [1,2], nums2 = [3,4]
输出:2.50000
解释:合并数组 = [1,2,3,4] ,中位数 (2 + 3) / 2 = 2.5
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
(1)解题思路1:
其实,我们不需要将两个数组真的合并,我们只需要找到中位数在哪里就可以了。
开始的思路是写一个循环,然后里边判断是否到了中位数的位置,到了就返回结果,但这里对偶数和奇数的分类会很麻烦。当其中一个数组遍历完后,出了 for 循环对边界的判断也会分几种情况。总体来说,虽然复杂度不影响,但代码会看起来很乱。
首先是怎么将奇数和偶数的情况合并一下。
用 len 表示合并后数组的长度,如果是奇数,我们需要知道第 (len+1)/2 个数就可以了,如果遍历的话需要遍历 int(len/2 ) + 1 次。如果是偶数,我们需要知道第 len/2和 len/2+1 个数,也是需要遍历 len/2+1 次。所以遍历的话,奇数和偶数都是 len/2+1 次。
返回中位数的话,奇数需要最后一次遍历的结果就可以了,偶数需要最后一次和上一次遍历的结果。所以我们用两个变量 left 和 right,right 保存当前循环的结果,在每次循环前将 right 的值赋给 left。这样在最后一次循环的时候,left 将得到 right 的值,也就是上一次循环的结果,接下来 right 更新为最后一次的结果。
循环中该怎么写,什么时候 A 数组后移,什么时候 B 数组后移。用 aStart 和 bStart 分别表示当前指向 A 数组和 B 数组的位置。如果 aStart 还没有到最后并且此时 A 位置的数字小于 B 位置的数组,那么就可以后移了。也就是aStart<m&&A[aStart]< B[bStart]。
但如果 B 数组此刻已经没有数字了,继续取数字 B[ bStart ],则会越界,所以判断下 bStart 是否大于数组长度了,这样 || 后边的就不会执行了(这里由于||短路特性,||左边判别式为真时,右边就自动不执行),也就不会导致错误了,所以增加为 (aStart<n)&&(bStart>= m || A[aStart]<B[bStart])
实现代码:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size();
int m = nums2.size();
int len = n + m;
int left = -1, right = -1;//right 保存当前循环的结果,left将得到 right 的值,也就是上一次循环的结果
int aStart = 0, bStart = 0;
for(int i = 0; i <= len / 2; i++){
left = right;
if((aStart < n) && (bStart >= m || nums1[aStart] < nums2[bStart])){
right = nums1[aStart++];
}else{
right = nums2[bStart++];
}
}
if((len & 1) == 0){//按位做与运算,看其最后一位是0还是1,亦即len是偶数还是奇数
return (left + right) / 2.0 ;
}else{
return right;
}
}
};
时间复杂度:遍历 len/2+1 次,len=m+n,所以时间复杂度依旧是 O(m+n)。
空间复杂度:我们申请了常数个变量,也就是 m,n,len,left,right,aStart,bStart 以及 i。
总共 8 个变量,所以空间复杂度是 O(1)。
(2)解题思路2:
用一个临时数组temp把num1和num2合并起来,再对temp进行排序,如果temp里元素个数len为奇数则返回temp[len/2],否则,返回。
实现代码:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size();
int m = nums2.size();
int len = n + m;
vector<int> temp(len);
merge(nums1.begin(),nums1.end(),nums2.begin(),nums2.end(),temp.begin());//合并两个数组并存入temp
sort(temp.begin(), temp.end());//对temp进行排序
if((len & 1) == 1){//奇数个元素
return temp[len/2];
}else{
return (temp[len/2 - 1] + temp[len/2]) / 2.0;
}
}
};