标签：top sp 中位数 神器 leftNum bp mp push

对顶堆——中位数神器

对顶堆，即一个大根堆和一个小根堆组合而成的一个数据结构，可以很方便的维护可变区间中位数。

平衡值

如果固定序列的大小为 n n n，那么平衡值指堆的最大大小。如果 n n n为偶数，大根堆的平衡值 l e f t N u m leftNum leftNum等于小根堆的平衡值 r i g h t N u m rightNum rightNum等于 n 2 \frac{n}{2} 2n​。如果 n n n为奇数，则 l e f t N u m = n / 2 leftNum = n/2 leftNum=n/2（整除）， r i g h t N u m = n − l e f t N u m rightNum = n - leftNum rightNum=n−leftNum。（不一定这么算，也可以让rightNum大一）

入堆

1. 检查左侧大根堆是否满足平衡要求，如果不满足平衡（ b q . s i z e ( ) < l e f t N u m bq.size() \lt leftNum bq.size()<leftNum）要求则直接插入大根堆。
2. 如果满足平衡要求，将当前元素和大根堆对顶元素比较，如果比对顶元素大或等于，则应该放在小根堆里面；如果比对顶元素小，则说明应该放进小根堆里面，并且对顶元素应该出堆，放进小根堆里面。

出堆

因为C++的std::priority_queue不支持删除元素，所以我们只能标记这个元素处于删除状态，并判这个元素属于哪一个堆，将该堆的平衡值加一，例如在大根堆里面则 l e f t N u m + 1 leftNum+1 leftNum+1。

这之后，如果有元素入堆，则应该先检查对顶元素是否为删除元素，如果是，直接将这个元素出堆，并调整平衡值，再按照平衡值调整正确的堆大小即可。

取中位数

如果 n n n为偶数，那么中位数就是大根堆堆顶元素加上小根堆堆顶元素的一半；如果是奇数，则应该按照平衡值去取。

例题

LeetCode 480 滑动窗口的中位数

class Solution
{
public:
    vector<double> medianSlidingWindow(vector<int> &nums, int k)
    {
        vector<double> ans;
        if (k == 1)
        {
            for (int i : nums)
            {
                ans.push_back(i);
            }
            return ans;
        }

        priority_queue<int> bp;
        priority_queue<int, vector<int>, greater<int>> sp;

        unordered_map<int, int> mp;

        int leftNum = k / 2;
        int rightNum = k - leftNum;

        for (int i = 0; i < k; i++)
        {
            if (bp.size() < leftNum)
                bp.push(nums[i]);
            else
            {
                if (bp.top() > nums[i])
                {
                    sp.push(bp.top());
                    bp.pop();
                    bp.push(nums[i]);
                }
                else
                {
                    sp.push(nums[i]);
                }
            }
        }

        if (k % 2 == 0)
        {
            ans.push_back((((double)bp.top()) + sp.top()) / 2);
        }
        else
        {
            ans.push_back(sp.top());
        }

        for (int i = 1, j = k; j < nums.size(); i++, j++)
        {
            mp[nums[i - 1]]++;

            if (nums[i - 1] <= bp.top())
            {
                leftNum++;
            }
            else
            {
                rightNum++;
            }

            while (!bp.empty() && mp.count(bp.top()))
            {
                mp[bp.top()]--;
                if (mp[bp.top()] == 0)
                {
                    mp.erase(bp.top());
                }
                bp.pop();
                leftNum--;
            }

            while (!sp.empty() && mp.count(sp.top()))
            {
                mp[sp.top()]--;
                if (mp[sp.top()] == 0)
                {
                    mp.erase(sp.top());
                }
                sp.pop();
                rightNum--;
            }

            while (bp.size() < leftNum && !sp.empty())
            {
                if (mp.count(sp.top()))
                {
                    mp[sp.top()]--;
                    if (mp[sp.top()] == 0)
                    {
                        mp.erase(sp.top());
                    }
                    sp.pop();
                    rightNum--;
                }
                else
                {
                    bp.push(sp.top());
                    sp.pop();
                }
            }

            while (!sp.empty() && mp.count(sp.top()))
            {
                mp[sp.top()]--;
                if (mp[sp.top()] == 0)
                {
                    mp.erase(sp.top());
                }
                sp.pop();
                rightNum--;
            }

            if (bp.size() < leftNum)
                bp.push(nums[j]);
            else
            {
                if (bp.top() > nums[j])
                {
                    sp.push(bp.top());
                    bp.pop();
                    bp.push(nums[j]);
                }
                else
                {
                    sp.push(nums[j]);
                }
            }

            while (!bp.empty() && mp.count(bp.top()))
            {
                mp[bp.top()]--;
                if (mp[bp.top()] == 0)
                {
                    mp.erase(bp.top());
                }
                bp.pop();
                leftNum--;
            }
            if (k % 2 == 0)
            {
                ans.push_back((((double)bp.top()) / 2 + ((double)sp.top()) / 2));
            }
            else
            {
                ans.push_back(sp.top());
            }
        }
        return ans;
    }
};

标签：top,sp,中位数,神器,leftNum,bp,mp,push
来源： https://blog.csdn.net/jiahonghao2002/article/details/114108760

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。