数据流中的中位数

 

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

分析：使用优先级队列将数据构建大堆（左边），小堆（右边），并且大堆中的所有数据都要小于小堆中的数据，并且两个堆中的数据一样，支撑这个过程的方法是奇数个数据时先入右边，然后将右边的top()入到左边，当数据时偶数个时，先入左边，再将左边的top()入到右边，这样既保证了右边的所有数据都大于左边的，又保证了数据平分左右两边，还保证了左边是大堆，右边是小堆。这样取中位数时，当此时数据有奇数个时，取左边的堆顶（因为第一次起始是先入到左边的），当数据个数是偶数时，就取两个堆顶的平均数。

//数据成员
    priority_queue<int, vector<int>, less<int>> left;//左边的大堆
    priority_queue<int, vector<int>, greater<int>> right;//右边的小堆
    int count=0;//数据的个数
    
    void Insert(int num)
    {
        count++;
        
        if(count%2==0)//偶数个数据
        {
            left.push(num);
            right.push(left.top());
            left.pop();
        }
        else//奇数个数据
        {
            right.push(num);
            left.push(right.top());
            right.pop();
        }
    }

    double GetMedian()
    { 
        if((count&1)==1)//奇数个
        {
            return left.top();
        }
        else
        {
            return (left.top()+right.top())/2.0;
        }
    }