题解 | #多数组中位数#

多数组中位数

http://www.nowcoder.com/practice/b6bb0bce88894108bfc23e9b7b012420

题目的主要信息:

给定两个升序的数组 arr1 和 arr2 ,求两个数组合并后的下中位数

注意:下中位数指在两个数组的数个数在偶数时取更小的

方法一:

先合并两个数组,然后再对合并后的数组排序,计算中位数的位置,最后直接返回中位数。

具体做法:

class Solution {
public:
    int getUpMedian(vector<int>& arr1, vector<int>& arr2) {
        for(int i=0;i<arr2.size();i++){//合并
            arr1.push_back(arr2[i]);
        }
        sort(arr1.begin(),arr1.end());//排序
        int n=arr1.size();//中位数位置
        return arr1[n/2-1];//返回中位数
    }
};

复杂度分析:

  • 时间复杂度:O(nlog2n)O(nlog_2n),sort排序的时间复杂度为O(nlog2n)O(nlog_2n)
  • 空间复杂度:O(1)O(1),只用了常数空间。

方法二:

首先计算两个数组的中位数位置target,同时遍历两个数组,由于两个数组已经是升序排序的,所以每次我们两数相比取更小的,直到取到第target个数就结束循环。需要注意的是,如果在遍历过程中,有数组提前结束遍历,则在另一个数组中继续。 alt 具体做法:

class Solution {
public:
    int getUpMedian(vector<int>& arr1, vector<int>& arr2) {
        int i = 0;//遍历arr1
        int j = 0;//遍历arr2
        int count = 0;
        int res = 0;
        int target = (arr1.size()+arr2.size())/2;
        while(i < arr1.size() || j < arr2.size()) {//遍历两个数组
            if (count == target) {//如果找到第target个数则结束循环
                break;
            }
            if (i == arr1.size()) {//arr1遍历完了则在arr2数组中继续
                res = arr2[j++];
            } else if (j == arr2.size()) {//arr2遍历完了则在arr1数组中继续
                res = arr1[i++];
            } else {//两个数组都没有遍历完
                if (arr1[i] > arr2[j]) {//取其更小的数
                    res = arr2[j++];
                } else {
                    res = arr1[i++];
                }
            }
              
            count++;
        }
        return res;
    }
};

复杂度分析:

  • 时间复杂度:O(n)O(n),while循环遍历两个数组。
  • 空间复杂度:O(1)O(1),只用了常数空间,
全部评论

相关推荐

网安已死趁早转行:山东这地方有点说法
点赞 评论 收藏
分享
01-24 12:50
门头沟学院 C++
投票
菜狗二号:还有啥想的 指定国有行啊,去了就开始幸福美满的生活了,选华子不是折腾自己么,最终财富积累度是差不多的,但是幸福指数是相差甚远的
点赞 评论 收藏
分享
评论
1
1
分享

创作者周榜

更多
牛客网
牛客企业服务