数据结构-单调队列

单调队列经常配合滑动窗口,求滑动中动态的各窗口区间内元素最值类问题。

239. 滑动窗口最大值

// 滑动窗口 + 单调队列(单调递减队列)

// 思路1:单调递减队列中存储数组元素下标

 // 窗口滑动时,当前元素若大于(或大于等于)队尾下标对应的值则循环出队尾, 若当前队首下标在窗口外(队首下标小于等于left=i-k)则循环队首出队 

class Solution {
public:
    // 滑动窗口 + 单调队列(单调递减队列)
    // 思路:
    // 单调递增队列中存储数组下标, 下标是递增的, 下标对应的值是递减的
    // 窗口滑动时,当前元素若大于(或大于等于)队尾下标对应的值则循环出队尾, 若当前队首下标在窗口外(队首下标小于等于left=i-k)则循环队首出队 
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        if (nums.size() * k == 0) {
            return {};
        }
        vector<int> res;
        deque<int> q;
        for (size_t i = 0; i < k; i++) {
            while (!q.empty() && nums[q.back()] < nums[i]) {
                q.pop_back();
            }
            q.push_back(i);
        }
        res.push_back(nums[q.front()]);
        for (size_t i = k; i < nums.size(); i++) {
            // 队列头是上一步窗口内最大值的索引值。若队列头不在当前窗口内,则队首出队. 循环比较
            while (!q.empty() && q.front() <= i - k) {
                q.pop_front();
            }
            while (!q.empty() && nums[q.back()] < nums[i]) {
                q.pop_back();
            }
            q.push_back(i);
            res.push_back(nums[q.front()]);
        }
        return res;
    }
};

// 思路2:单调递减队列中存储数组元素值

// 实现单调递增队列
class MonotonicQueue {
public:
    void push(int n) {
        // 将小于 n 的元素全部删除
        while (!q.empty() && q.back() < n) {
            q.pop_back();
        }
        // 然后将 n 加入尾部
        q.push_back(n);
    }
    int max() {
        return q.front();
    }
    
    void pop(int n) {
        if (n == q.front()) {
            q.pop_front();
        }
    }
private:
    deque<int> q;
};

class Solution1 {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        if (nums.size() * k == 0) {
            return {};
        }
        MonotonicQueue window;
        vector<int> res;

        for (int i = 0; i < nums.size(); i++) {
            if (i < k - 1) {
                //先填满窗口的前 k - 1
                window.push(nums[i]);
            } else {
                // 窗口向前滑动,加入新数字(在这里之前window中元素个数最多是k-1个,本步骤是加上新尾)
                window.push(nums[i]);
                // 记录当前窗口的最大值(上面一行加上新尾后,window中元素个数最多是k个)
                res.push_back(window.max());
                // 移出旧数字(去旧头)
                window.pop(nums[i - k + 1]);
            }
        }
        return res;
    }
};

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值