问题
思路
这个题目的意思是,数据流中一直有数据在插入,怎么样实时返回中位数。这个题目还是比较有实际意义,因为像在线系统一般都是实时插入数据。
这个题目是借助辅助数据结构,堆去完成。
对于一个数据,借助两个堆来进行排序,并且两者的元素之差始终不超过1个。
这个题目还是比较有技巧的!
bug也改了我好长时间,就是要小心max_heap上来为空的情形,因此此时不用判断min_heap是否为空。它肯定为空。
每个堆插入元素的时候都需要判断是否破坏了整个数组的有序性
代码
#include <iostream>
#include <queue>
using namespace std;
class Solution {
private:
priority_queue< int, vector<int>, less<int> > max_heap; // less element
priority_queue< int, vector<int>, greater<int> > min_heap; // greater element
public:
void Insert(int num)
{
int sz = max_heap.size() + min_heap.size();
if( sz%2 ){
// add num to min_heap
if( num > max_heap.top() ) min_heap.push(num);
else{
max_heap.push(num);
num = max_heap.top();
max_heap.pop();
min_heap.push( num );
}
}else{
// add num to max_heap
if( max_heap.empty() ) {
max_heap.push(num);
return;
}
if( num < min_heap.top() ) max_heap.push( num );
else{
min_heap.push( num );
num = min_heap.top();
min_heap.pop();
max_heap.push( num );
}
}
}
double GetMedian()
{
int sz1 = max_heap.size();
int sz2 = min_heap.size();
if( sz1 == sz2 ) return 1.0/2*( max_heap.top() + min_heap.top() );
else if( sz1 + 1 == sz2 ){
return min_heap.top();
}
else return max_heap.top();
}
};