剑指OFFER 数据流中的中位数

Posted 2021-03-11 virgildevil

剑指OFFER 数据流中的中位数

题目描述

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

代码(stl multiset)

class Solution {
public:
    multiset<int> m;
    int count = 0;

    void Insert(int num)
    {
        m.insert(num);
        count++;
    }

    double GetMedian()
    {
        if(count%2==0)
        {//even
            int i = 0;
            for(auto it = m.begin();it!=m.end();it++)
            {
                if(i==count/2)
                {
                    int x = *it;
                    it--;
                    int y = *it; 
                    return (x+y)/2.0;
                }
                
                i++;
            }
        }else{//odd
            int i = 0;
            for(auto it = m.begin();it!=m.end();it++)
            {
                if(i==count/2)
                {
                    return *it;
                }
                i++;
            }
        }
    }

};