难度困难
给定两个大小分别为 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
本人通过代码(附注释):
#include<iostream>
#include<vector>
using namespace std;
// 快速排序
void qsort(vector<int>& nums,int low,int high){
if(low >= high) return ;
int key = nums[low]; // 将第一个元素作为关键字
int i = low;
int j = high;
while(i < j){ // 当i>=j时跳出循环
while(i < j && nums[j] >= key) j--; // 从右向左找第一个小于key的值
if(i < j) nums[i++] = nums[j];
while(i < j && nums[i] <= key) i++; // 从左向右找第一个大于key的值
if(i < j) nums[j--] = nums[i];
}
nums[i] = key; // 将key放到正确位置
qsort(nums, low, i-1); // 递归调用
qsort(nums, j+1, high);
}
// 查找两个已排序数组的中位数
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
double midnum;
int len;
for(int i=0;i<nums2.size();i++){
nums1.push_back(nums2[i]); // 将nums2中的元素加到nums1末尾
}
qsort(nums1,0,nums1.size()-1); // 对新的nums1进行快速排序
len = nums1.size();
if(len%2 == 1){
midnum = double(nums1[(len-1)/2]); // 如果元素个数为奇数,则返回中间的值
}
else{
midnum = (nums1[(len)/2] + nums1[(len-1)/2]) / 2.0; // 如果元素个数为偶数,则返回中间两个数的平均值
}
return midnum;
}
int main(){
vector<int> nums1;
vector<int> nums2;
int num;
char c1,c2;
while(c1 != '\n'){
cin >> num;
nums1.push_back(num); // 输入nums1数组
c1 = getchar();
}
while(c2 != '\n'){
cin >> num;
nums2.push_back(num); // 输入nums2数组
c2 = getchar();
}
cout << findMedianSortedArrays(nums1,nums2); // 输出中位数
return 0;
}