class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int length1 = nums1.size();
int length2 = nums2.size();
//int mid1 = length1/2;
//int mid2 = length2/2;
vector<int> nums3;
int i=0, j=0;
if(length1 == 0)
nums3 = nums2;
else if(length2 == 0)
nums3 = nums1;
else{
while(i<length1 && j<length2){
if(nums1[i]>nums2[j]){
nums3.push_back(nums2[j]);
j++;
}
else if(nums1[i]<nums2[j]){
nums3.push_back(nums1[i]);
i++;
}
else{
nums3.push_back(nums1[i]);
i++;
nums3.push_back(nums2[j]);
j++;
}
}
while(i<length1) {
nums3.push_back(nums1[i]);
i++;
}
while(j<length2) {
nums3.push_back(nums2[j]);
j++;
}
}
int length=nums3.size();
if(length%2 == 0)
return (nums3[length/2]+nums3[length/2-1])/2.0;
else
return (nums3[(length-1)/2]);
}
};
采用的方法为先将两个数组合并,然后选中位数,时间复杂度为O(m+n),后续考虑算法要求的O(log(m+n))的办法。
leetcode-04-两个排序数组的中位数
最新推荐文章于 2020-08-09 21:01:18 发布