剑指offer 数据流中的中位数

题目描述

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

 

 

思路：利用两个优先队列，最大堆始终保存前一半数据，最小堆始终保存后一半数据，当两个优先队列长度一样时，将数压入最小堆。

 1 class Solution {
 2     priority_queue<int, vector<int>, less<int> > p; //最大堆
 3     priority_queue<int, vector<int>, greater<int> > q; //最小堆
 4 public:
 5     void Insert(int num)
 6     {    
 7         //偶数个，应该压入最小堆
 8         if (((p.size() + q.size()) & 1) == 0) {
 9             if (p.size() > 0 && num < p.top()) {
10                 int temp = p.top();
11                 p.pop();
12                 p.push(num);
13                 q.push(temp);
14             } else {
15                 q.push(num);
16             }
17         } else { //奇数个，应该压入最大堆，但是数可能比最小堆的最小值大
18             if (q.size() > 0 && num > q.top()) {
19                 int temp = q.top();
20                 q.pop();
21                 q.push(num);
22                 p.push(temp);
23             } else {
24                 p.push(num);
25             }
26         }
27     }
28 
29     double GetMedian()
30     { 
31         if (p.size() == q.size()) {
32             return (p.top() + q.top()) / 2.0;
33         } else {
34             return q.top();
35         }
36     
37     }
38 
39 };

 

来源：https://www.cnblogs.com/qinduanyinghua/p/11403207.html