C++ 堆heap 寻找中位数

设计一个数据结构,该数据结构动态维护一组数据,且支持如下操作:

  1. 添加元素:void addNum(int num),将整型num添加至数据结构中。
  2. 返回数据的中位数:double findMedian(),返回其维护的数据的中位数。
    中位数定义:
  3. 若数据个数为奇数,中位数是该数组排序后中间的数。
  4. 若数据个数为偶数,中位数是该数据排序后中间的两个数字的平均值。

最直观的方法:
存储结构使用数组,每次添加元素或查找中位数时对数组排序,再计算结果。
时间复杂度:

  1. 若添加元素时排序,addNum复杂度O(n), findMedian复杂度O(1)
  2. 若查询中位数时排序,addNum复杂度O(1), findMedian复杂度O(nlogn)
    若添加元素或查询中位数时随机的操作,共n次,按上述思想,整体复杂度最佳为O(n^2)

巧用堆的性质:
动态维护一个最大堆与一个最小堆,最大堆存储一半数据,最小堆存储一半数据,维持最大堆的堆顶比最小堆的堆顶小。

#include<stdio.h>
#include<queue>
class MedianFinder
{
   
public:
 MedianFinder(){
   }
 ~MedianFinder(){
   }
 std::priority_queue<int, std::vector<int>, std::greater<int>> small_queue;    //最小堆
 std::priority_queue<int> big_queue;             //最大堆
 void addNum(int num)
 {
   
  if (big_queue.empty())
  
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值