【剑指offer】数据流中的中位数

题目描述

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

链接

https://www.nowcoder.com/practice/9be0172896bd43948f8a32fb954e1be1?tpId=13&tqId=11216&rp=1&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

代码

class Solution {
public:
	vector<int> max;  // 最大值，大顶堆 
	vector<int> min;  // 最小值，小顶堆 
    void Insert(int num)
    {
    	int len = max.size() + min.size();
        if(len%2 == 0){  // 目前有偶数个数据，将当前数据插入至小顶堆
        	if(min.size() > 0 && num > min[0]){
        		int temp = min[0];
        		min.push_back(num);
        		pop_heap(min.begin(),min.end(),greater<int>());
        		max.push_back(temp);
        		push_heap(max.begin(),max.end(),less<int>());
        		min.pop_back();
        	} 
        	else{
        		max.push_back(num);
        		push_heap(max.begin(),max.end(),less<int>());
        	}
        }
        else{
        	if(max.size() > 0 && num < max[0]){
        		int temp = max[0];
        		max.push_back(num);
        		pop_heap(max.begin(),max.end(),less<int>());
        		min.push_back(temp);
        		push_heap(min.begin(),min.end(),greater<int>());
        		max.pop_back();
        	}
        	else{
        		min.push_back(num);
        		push_heap(min.begin(),min.end(),greater<int>());
        	}
        }
    }

    double GetMedian()
    { 
    	int len = max.size() + min.size();
    	if(len <= 0){
    		return 0;
    	}
    	if(len%2 == 0){
    		return ((max[0] + min[0]) / 2.0);
    	}
    	else{
    		return max[0];
    	}
    }
};