对顶堆+延迟删除(lc.480)

seventeena

已于 2024-01-23 16:35:48 修改

阅读量530

点赞数 11

分类专栏： leetcode刷题日志文章标签：算法 c++ 数据结构

于 2024-01-22 11:38:19 首次发布

本文链接：https://blog.csdn.net/seventeena/article/details/135743931

版权

leetcode刷题日志专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文介绍了如何使用优先队列（双堆）数据结构解决求多组数的滑动窗口中位数问题，包括插入、删除操作以及维护堆的平衡，同时提及了与之相关的ACwing题目作为相似题型参考。

摘要由CSDN通过智能技术生成

1.题目

2.适用范围

一般适用于求多组数的中位数
根据题目要求，中位数是down.top()或者是(up.top() + down.top()) / 2
其它细节都在代码注释里

3.数据结构模型

4.代码如下

class DualHeap {
    private:
        // 下堆是大根堆
        priority_queue<int> down;
        // 上堆是小根堆
        priority_queue<int, vector<int>, greater<int>> up;
        // 延迟删除堆中元素,只有当其在堆顶的时候才进行删除
        // [k, v] 分别代表[要删除的数值、出现的次数]
        unordered_map<int, int> delayed;
        // 窗口大小(题目要求)
        int k;
        // 这里十分重要,调整平衡的时候不是比较虚拟大小(down.size),而是比较真实大小
        // 记录上下堆的实际大小
        int downSize, upSize;

    public:
        DualHeap(int _k): k(_k), downSize(0), upSize(0) {}
    
    private:
        // 延迟删除(修剪)
        // 如果堆顶元素是要删除的元素,则一直删除,直到堆顶元素在滑窗内
        void prune(auto &heap) {
            while (heap.size()) {
                int x = heap.top();
                if (delayed.count(x)) {
                    if (-- delayed[x] == 0)
                        delayed.erase(x);
                    heap.pop();
                }
                else break;
            }
        }
    
        // 维持两个堆的大小
        /** 维持条件:
            1.下堆至多大于上堆1个,至少与上堆个数相等
        	2.min(上堆) >= max(下堆)
        **/
        void makeBalance() {
            if (downSize > upSize + 1) {s
                up.push(down.top());
                down.pop();
                -- downSize, ++ upSize;
                prune(down);
            }
            else if (downSize < upSize) {
                down.push(up.top());
                up.pop();
                ++ downSize, -- upSize;
                prune(up);
            }
        }
    public:
        // 插入
        void insert(int num) {
            if (down.empty() || num <= down.top()) down.push(num), ++ downSize;
            else up.push(num), ++ upSize;

            makeBalance();
        }
        // 删除
        void erase(int num) {
            ++ delayed[num];
            if (num <= down.top()) {
                -- downSize;
                if (num == down.top())
                    prune(down);
            }
            else {
                -- upSize;
                if (num == up.top())
                    prune(up);
            }

            makeBalance();
        }
        // 求中位数(题目要求)
        double getMedian() {
            return k & 1 ? down.top() : ((double)down.top() + up.top()) / 2;
        }
};