剑指offer:数据流中的中位数

描述

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

数据范围：数据流中数个数满足 1 \le n \le 1000 \1≤n≤1000  ，大小满足 1 \le val \le 1000 \1≤val≤1000 

进阶： 空间复杂度 O(n) \O(n)  ， 时间复杂度 O(nlogn) \O(nlogn) 

示例1

输入：

[5,2,3,4,1,6,7,0,8]

复制返回值：

"5.00 3.50 3.00 3.50 3.00 3.50 4.00 3.50 4.00 "

复制说明：

数据流里面不断吐出的是5,2,3...,则得到的平均数分别为5,(5+2)/2,3...   

思路:

1.使用插入排序

2.返回中位数就可以直接计算

class Solution {
public:
    vector<int> data;
    void Insert(int num) {
        if(data.empty())data.push_back(num);
        else{
            auto it = lower_bound(data.begin(), data.end(), num);
            data.insert(it, num);
        }
    }

    double GetMedian() { 
        int size=data.size();
        if(size %2==0){
            double res=static_cast<double>(data[size/2-1]+data[size/2])/2;
            return res;
        }else{
            return static_cast<double>(data[size/2]);
        }
    }

};