4. 寻找两个正序数组的中位数
给定两个大小分别为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
解题思路:
我的第一遍代码提交,实现了功能。但是看完官方题解以后,我感觉我是来捣乱的。
1.不能使用std已经实现的方法。vector的insert()和sort()。
2.中位数是指1/2长度的那位数。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
nums1.insert(nums1.end(),nums2.begin(),nums2.end());
std::sort(nums1.begin(),nums1.end());
int size = nums1.size();
if(0==size){
return 0;
}else if(1==size){
return nums1[0];
}else{
int mid = size/2;
if (0 != size%2){
return nums1[mid];
}else{
double sum = nums1[mid-1]+nums1[mid];
return sum/2;
}
}
}
};
一开始还疑惑为什么这道题目,难度是困难
。这道题目有两种解题思路,并且考察了不同的知识点。
-
思路一:使用
归并
的方式,先将2个数组合并,然后从1个大的数组中查找出中位数。也就是我的解题思路,但是我直接使用了std的方法,就有点儿不要脸了。假设nums1的长度是m,nums2的长度是n。那么该思路的空间复杂度是O(m+n),时间复杂度是O(m+n)。 -
因为nums1和nums2的长度是已知的,那么可知:
- 如果m+n是奇数的话,
中位数必然是(m+n)/2
。 - 如果m+n是偶数的话,中位数是
下标(m+n)/2
和下标(m+n)/2-1
所对应的元素之和的一半
。
思路二:声明2个指针,分别指向两个数组,比较2个指针指向的元素谁更小,谁就往后移一步,并且自增计数。当m+n是奇数时,计数等于(m+n)/2即为中位数。当m+n是偶数时,取下标
(m+n)/2和下标(m+n)/2-1元素和的一半
。 - 如果m+n是奇数的话,