力扣295. 数据流的中位数

优先队列

  • 思路:
    • 中位数是排序中间的数值:S1.M.S2
    • 可以使用两个优先队列来存放两边的数值,总是使得左侧的堆顶是最大的,右侧的堆顶是最小的,即使用大顶堆存放 S1,使用小顶堆存放S2,使得两个队列的 size 维持"平衡",则中位数就会在两个堆顶"附近"了;
    • 维持两个队列 size 平衡:
      • 数据先 push 的大顶堆,如果是 > M 的数,则会在堆顶;如果是 < M 的数,则会沉入队列中;
      • 然后将堆顶的数 push 到小顶堆,如果是 > M 的数,会沉入队列;如果是 < M 的数,会在堆顶;
      • 将大顶堆的堆顶 pop;(因为已经 push 到小顶堆)
      • 判断一下两个队列的size,如果大顶堆的 size 少了,将小顶堆的堆顶"漏"到大顶堆;(可以将两个队列组合成漏斗,更直观)
    • 此时的中位数:
      • 如果大顶堆 size 多,则中位数是其堆顶;
      • 否则,为两个堆顶的均值;
cpp 复制代码
class MedianFinder {
public:
    MedianFinder() {

    }
    
    void addNum(int num) {
        low.push(num);
        high.push(low.top());
        low.pop();

        if (low.size() < high.size()) {
            low.push(high.top());
            high.pop();
        }
    }
    
    double findMedian() {
        if (low.size() > high.size()) {
            return low.top();
        }

        return (low.top() + high.top()) / 2.0;
    }

private:
    std::priority_queue<int, std::vector<int>, std::less<int>> low;
    std::priority_queue<int, std::vector<int>, std::greater<int>> high;
};

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder* obj = new MedianFinder();
 * obj->addNum(num);
 * double param_2 = obj->findMedian();
 */
相关推荐
黄焖鸡能干四碗3 分钟前
MES生产执行制造系统建设(Java+Mysql)
java·大数据·开发语言·信息可视化·需求分析
chao1898448 分钟前
基于MATLAB的双摆系统阻抗控制实现
算法
舒克日记11 分钟前
基于springboot的民谣网站的设计与实现
java·spring boot·后端
龙腾AI白云1 小时前
大模型-AIGC技术在文本生成与音频生成领域的应用
算法
凤山老林2 小时前
新一代Java应用日志可视化与监控系统开源啦
java·后端·开源
流星白龙2 小时前
【Qt】7.信号和槽_connect函数用法(2)
java·数据库·qt
坚持编程的菜鸟5 小时前
LeetCode每日一题——困于环中的机器人
c语言·算法·leetcode·机器人
黎宇幻生6 小时前
Java全栈学习笔记39
java·笔记·学习
Aurorar0rua7 小时前
C Primer Plus Notes 09
java·c语言·算法
nongcunqq8 小时前
abap 操作 excel
java·数据库·excel