480. 滑动窗口中位数

最新推荐文章于 2024-10-07 09:58:35 发布

qq_43983809

最新推荐文章于 2024-10-07 09:58:35 发布

阅读量62

点赞数

文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_43983809/article/details/134143324

版权

480. 滑动窗口中位数

在这里插入图片描述

题目比较有意思记录一下做题思路
用到c++中的multiset数据结构，和set不同的是multiset可以存相同元素
维护两个multiset，queMax和queMin，queMin记录小于等于中位数的数据，queMax记录大于中位数的数据，实现add和del两个函数，当前数小于等于queMin的最大值，插入queMin，否则插入queMax，同时需要维护queMin的size等于插入总数cnt/2或者cnt/2+1

class Solution {
public:
    multiset<int, greater<int>> queMin;
    multiset<int>queMax;
    void add(int num){
        if(queMin.size() == 0|| *queMin.begin()>= num){
            queMin.insert(num);
            if(queMin.size() > queMax.size() + 1){
                queMax.insert(*queMin.begin());
                queMin.erase(queMin.begin());
            }
        }else{
            queMax.insert(num);
            if(queMax.size() > queMin.size()){
                queMin.insert(*queMax.begin());
                queMax.erase(queMax.begin());
            }
        }
    }
    void det(int num){

        if(queMin.find(num)!= queMin.end()){
            queMin.erase(queMin.find(num));
            if(queMin.size()<queMax.size()){
                queMin.insert(*queMax.begin());
                queMax.erase(queMax.begin());
            }
        }else{
            queMax.erase(queMax.find(num));
            if(queMax.size()<int(queMin.size()) - 1){
                queMax.insert(*queMin.begin());
                queMin.erase(queMin.begin());
            }
        }
    }
    vector<double> medianSlidingWindow(vector<int>& nums, int k) {
        int n = nums.size();
        vector<double> ans;
        for(int i=0;i<k;i++){
            add(nums[i]);
        }
        if(k&1){
            ans.push_back(*queMin.begin());
        }else{
            ans.push_back((double(*queMax.begin()) + *queMin.begin())/2);
        }
        for(int i=k;i<n;i++){
            add(nums[i]);
            det(nums[i-k]);
            if(k&1){
                ans.push_back(*queMin.begin());
            }else{
                ans.push_back((double(*queMax.begin()) + *queMin.begin())/2);
            }
        }
        return ans;
    }
};