单调队列

最新推荐文章于 2024-09-05 22:54:21 发布

Vivian_345

最新推荐文章于 2024-09-05 22:54:21 发布

阅读量412

点赞数

分类专栏：单调队列文章标签：数据结构

本文链接：https://blog.csdn.net/weixin_41828075/article/details/116303327

版权

单调队列专栏收录该内容

1 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何利用双端队列（deque）数据结构解决滑动窗口最大值问题，给出了C++和Python两种语言的实现代码。在C++中使用了单调栈，而在Python中同样利用了队列的特性来保持滑动窗口内的最大值。这种方法的时间复杂度为O(n)，有效地解决了动态获取数组中固定大小窗口最大值的问题。

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

239. 滑动窗口最大值

给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回滑动窗口中的最大值。

输入：nums = [1,3,-1,-3,5,3,6,7], k = 3
输出：[3,3,5,5,6,7]
解释：
滑动窗口的位置最大值
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7

双端队列的操作：

class deque {
    // 在队头插入元素 n
    void push_front(int n);
    // 在队尾插入元素 n
    void push_back(int n);
    // 在队头删除元素
    void pop_front();
    // 在队尾删除元素
    void pop_back();
    // 返回队头元素
    int front();
    // 返回队尾元素
    int back();
}

C++ 单调栈：

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> res;
        deque<int> deq;
        for(int i=0; i<nums.size(); i++){
            if(!deq.empty() && deq.front()==i-k) deq.pop_front(); //控制队列的长度为k
            while(!deq.empty() && nums[deq.back()]<nums[i]) deq.pop_back(); //保持队首元素为最大值
            deq.push_back(i);
            if(i>=k-1) res.push_back(nums[deq[0]]);
        }
        return res;
    }
};

python：

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        res = []
        queue = []
        for i,num in enumerate(nums):
            if queue and queue[0] == i-k:
                queue.pop(0)
            while queue and nums[queue[-1]]<num:
                queue.pop()
            queue.append(i)
            if i>=k-1:
                res.append(nums[queue[0]])
        return res