首页 > 编程语言 > 详细

寻找两个有序数组的中位数

时间:2019-09-10 12:58:52      阅读:69      评论:0      收藏:0      [点我收藏+]

【问题】给定两个大小为 m 和 n 的有序数组nums1 和 nums2

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0
示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

【思路】比较简单的思路使用归并排序,为什么有这个思路呢?因为题目说两个数组是有序数组,因此我们对两个数组进行merge,如果小的数则放入res数组中,直到res的数组大小为(m*n)/2+1,因此最后在总个数为偶数时,中位数为res中最后两个数求平均,否则中位数为res数组中最后一个数。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        int m = nums1.size(), n = nums2.size();
        int mid = (m + n) / 2 + 1;
        int i, j;
        for(i = 0, j = 0;i <m && j < n;){
            if(nums1[i] > nums2[j]){
                res.push_back(nums2[j++]);
            }
            else if(nums1[i] <= nums2[j]){
                res.push_back(nums1[i++]);
            }
            if(res.size() == mid){
                break;
            }
        }
        while(res.size() != mid){
            if(i < m)
                res.push_back(nums1[i++]);
            if(j < n)
                res.push_back(nums2[j++]);
        }

        int len = res.size();
        if((m + n) % 2 == 0)
            return (res[len-1] + res[len-2]) / 2.0;
        return res[len-1];
    }
};

 

寻找两个有序数组的中位数

原文:https://www.cnblogs.com/zhudingtop/p/11496108.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!