LeetCode 寻找两个正序数组的中位数

4. 寻找两个正序数组的中位数

难度困难

给定两个大小分别为 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;
}