代码随想录算法训练营第十二天|栈与队列的内容总结

LeetCode239. 滑动窗口最大值

题目链接

https://leetcode.cn/problems/sliding-window-maximum/

题目分析

设计单调队列的时候,pop,和push操作要保持如下规则:

  1. pop(value):如果窗口移除的元素value等于单调队列的出口元素,那么队列弹出元素,否则不用任何操作

  1. push(value):如果push的元素value大于入口元素的数值,那么就将队列入口的元素弹出,直到push元素的数值小于等于队列入口元素的数值为止

保持如上规则,每次窗口移动的时候,只要问que.front()就可以返回当前窗口的最大值。

为了更直观的感受到单调队列的工作过程,以题目示例为例,输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3,动画如下:

代码如下:

class Solution {
private:
    class MyQueue {
    public:
        deque<int> que;//定义一个队列,实现单调队列的功能
        
        //pop操作,当要删除的值与队列出口的值相等时,将队列出口的值pop掉
        //否则,无需进行操作
        //注意,要实现确定队列不为空,以保证pop操作有效。
        void pop(int value) {
            if(!que.empty() && value == que.front()) {
                que.pop_front();
            }
        }

        //push操作,当要插入的值比队列入口的值大时,将队列入口的值弹出
        //循环进行,直到队列为空或者插入的值比队列入口的值小,此时将要插入的值插入
        //保证了队列中的元素是从大到小的顺序排列
        void push(int value) {
            while(!que.empty() && value > que.back()) {
                que.pop_back();
            }
            que.push_back(value);
        }
        
        //返回当前队列的出口元素,也就是最大值
        int refront() {
            return que.front();
        }
    };
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MyQueue que;
        vector<int> result;
        //事先把第一组窗口数值放入
        for(int i = 0; i < k; i++) {
            que.push(nums[i]);
        }
        //返回最大值
        result.push_back(que.refront());
        //窗口滑动,先pop再push,然后返回最大值
        for(int j = k; j < nums.size(); j++) {
            que.pop(nums[j - k]);
            que.push(nums[j]);
            result.push_back(que.refront());
        }

        return result;
    }
};

LeetCode

class Solution {

private:

class MyQueue {

public:

deque<int> que;//定义一个队列,实现单调队列的功能

//pop操作,当要删除的值与队列出口的值相等时,将队列出口的值pop掉

//否则,无需进行操作

//注意,要实现确定队列不为空,以保证pop操作有效。

void pop(int value) {

if(!que.empty() && value == que.front()) {

que.pop_front();

}

}

//push操作,当要插入的值比队列入口的值大时,将队列入口的值弹出

//循环进行,直到队列为空或者插入的值比队列入口的值小,此时将要插入的值插入

//保证了队列中的元素是从大到小的顺序排列

void push(int value) {

while(!que.empty() && value > que.back()) {

que.pop_back();

}

que.push_back(value);

}

//返回当前队列的出口元素,也就是最大值

int refront() {

return que.front();

}

};

public:

vector<int> maxSlidingWindow(vector<int>& nums, int k) {

MyQueue que;

vector<int> result;

//事先把第一组窗口数值放入

for(int i = 0; i < k; i++) {

que.push(nums[i]);

}

//返回最大值

result.push_back(que.refront());

//窗口滑动,先pop再push,然后返回最大值

for(int j = k; j < nums.size(); j++) {

que.pop(nums[j - k]);

que.push(nums[j]);

result.push_back(que.refront());

}

return result;

}

};

猜你喜欢

转载自blog.csdn.net/wyr1849089774/article/details/128764592
今日推荐