C++刷剑指offer（面试题41. 数据流中的中位数）【大小堆】

糖逗

题目描述：

1. 如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。
   
2. 
   
3. 例如，
   
4. 
   
5. [2,3,4] 的中位数是 3
   
6. 
   
7. [2,3] 的中位数是 (2 + 3) / 2 = 2.5
   
8. 
   
9. 设计一个支持以下两种操作的数据结构：
   
10. 
    
11. void addNum(int num) - 从数据流中添加一个整数到数据结构中。
    
12. double findMedian() - 返回目前所有元素的中位数。
    
13. 示例 1：
    
14. 
    
15. 输入：
    
16. ["MedianFinder","addNum","addNum","findMedian","addNum","findMedian"]
    
17. [[],[1],[2],[],[3],[]]
    
18. 输出：[null,null,null,1.50000,null,2.00000]
    
19. 示例 2：
    
20. 
    
21. 输入：
    
22. ["MedianFinder","addNum","findMedian","addNum","findMedian"]
    
23. [[],[2],[],[3],[]]
    
24. 输出：[null,null,2.00000,null,2.50000]
    
25.  
    
26. 
    
27. 限制：
    
28. 
    
29. 最多会对 addNum、findMedia进行 50000 次调用。
    
30. 
    
31. 来源：力扣（LeetCode）
    
32. 链接：https://leetcode-cn.com/problems/shu-ju-liu-zhong-de-zhong-wei-shu-lcof
    
33. 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

复制代码

1. #include <iostream>
   
2. #include <queue>
   
3. 
   
4. using namespace std;
   
5. 
   
6. class MedianFinder{
   
7. public:
   
8.         priority_queue<int, vector<int>, greater<int> > great;
   
9.         priority_queue<int, vector<int>, less<int> > less;
   
10.        
    
11.         void addNum(int num){
    
12.                 great.push(num);
    
13.                 less.push(great.top());
    
14.                 great.pop();
    
15.                
    
16.                 if(great.size() < less.size()){
    
17.                         great.push(less.top());
    
18.                         less.pop();
    
19.                 }
    
20.         }
    
21.        
    
22.        
    
23.         double findMedian(){
    
24.                 double res;
    
25.                 if(great.size() == less.size()) res = (great.top() + less.top()) * 0.5;
    
26.                 if(great.size() > less.size()) res = (double)great.top();
    
27.                 cout << res << endl;
    
28.                 return res;
    
29.         }
    
30. };
    
31. 
    
32. 
    
33. 
    
34. int main(void){
    
35.         MedianFinder test;
    
36.         test.addNum(3);
    
37.         test.addNum(2);
    
38.         test.findMedian();
    
39.         test.addNum(4);
    
40.         test.findMedian();
    
41.         return 0;
    
42. }

复制代码

注意事项：
1.参考链接：https://leetcode-cn.com/problems ... ian-dui-lie-by-z1m/
2.参考链接：https://blog.csdn.net/weixin_36888577/article/details/79937886
3.用大小堆找中位数