如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
class Solution { public: void Insert(int num) { data.push_back(num); std::sort(data.begin(), data.end()); } double GetMedian() { unsigned int size = data.size(); if (size & 1) { return data[size >> 1]; } else { int left = data[(size >> 1) - 1]; int right = data[size >> 1]; return (static_cast<double>(left) + right) / 2; } } private: vector<int> data; };
class Solution { private: vector<int> min; vector<int> max; public: void Insert(int num) { if(((min.size()+max.size())&1)==0)//偶数时 ,放入最小堆 { if(max.size()>0 && num<max[0]) { // push_heap (_First, _Last),要先在容器中加入数据,再调用push_heap () max.push_back(num);//先将元素压入容器 push_heap(max.begin(),max.end(),less<int>());//调整最大堆 num=max[0];//取出最大堆的最大值 //pop_heap(_First, _Last),要先调用pop_heap()再在容器中删除数据 pop_heap(max.begin(),max.end(),less<int>());//删除最大堆的最大值 max.pop_back(); //在容器中删除 } min.push_back(num);//压入最小堆 push_heap(min.begin(),min.end(),greater<int>());//调整最小堆 } else//奇数时候,放入最大堆 { if(min.size()>0 && num>min[0]) { // push_heap (_First, _Last),要先在容器中加入数据,再调用push_heap () min.push_back(num);//先压入最小堆 push_heap(min.begin(),min.end(),greater<int>());//调整最小堆 num=min[0];//得到最小堆的最小值(堆顶) //pop_heap(_First, _Last),要先调用pop_heap()再在容器中删除数据 pop_heap(min.begin(),min.end(),greater<int>());//删除最小堆的最大值 min.pop_back(); //在容器中删除 } max.push_back(num);//压入数字 push_heap(max.begin(),max.end(),less<int>());//调整最大堆 } } /*获取中位数*/ double GetMedian() { int size=min.size()+max.size(); if(size<=0) //没有元素,抛出异常 return 0;//throw exception("No numbers are available"); if((size&1)==0)//偶数时,去平均 return ((double)(max[0]+min[0])/2); else//奇数,去最小堆,因为最小堆数据保持和最大堆一样多,或者比最大堆多1个 return min[0]; } };
