239. 滑动窗口最大值(栈与队列)(Leetcode刷题笔记)
欢迎大家访问我的GitHub博客
https://lunan0320.cn
题目
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1
输出:[1]
解题代码 C++(核心代码)
解题思路: 本题目感觉还是需要技巧的,因此查看了题解。学会使用单调队列的方法。 学会使用栈和队列的底层容器 deque。
deque是一个双向队列,可以在队列的前后添加和删除元素。
方法:构造一个单调队列。队列的首部就是当前的最大值。是一个从大到小的队列。在窗口移动的过程中,新元素比队列尾部小,那就入队列,否则让队列的尾部出队列,直到队列为空或者队列尾部比新元素大。
在窗口移动的过程中,出队列的时候,考虑队列首元素是否和当前元素一样,一样的话就pop,否则就不操作(因为不一样的话,说明这个元素压根没有在队列中)
class Solution {
private:
//维护一个单调队列
class myQueue{
public:
deque<int> que;
//push操作
void push(int value){
//当队列中最后的元素比当前value小的时候,那就弹出
while (!que.empty() && que.back() < value){
que.pop_back();
}
//当前元素入队列
que.push_back(value);
}
//pop操作
void pop(int value){
//队列首部和value相同,说明需要删除
if (que.front() == value){
que.pop_front();
}
}
//取队列的首元素
int front(){
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> res;
myQueue que;
//相当于对que的初始化,前k个处理
for (int i = 0; i < k; i++){
que.push(nums[i]);
}
//前k个的最大值入res
res.push_back(que.front());
//窗口的移动
int i = k;
while (i < nums.size()){
//窗口移动 先pop
//注意此处pop的是窗口的最左端,索引是i-k
que.pop(nums[i-k]);
// 窗口移动 push进去
que.push(nums[i]);
// 此时单调队列首部就是最大的
res.push_back(que.front());
i++;
}
return res;
}
};
解题代码 C++(本地编译运行)
/* head.h */
#include <iostream>
#include <stack>
#include <string>
#include <vector>
using namespace std;
/* myStack.h */
#include "head.h"
//滑动窗口的最大值
class MyWindow {
private:
class myQueue {
public:
deque<int> que;
void push(int value) {
// 队列的尾部比value小的话,就把末尾元素pop。这样保证队列的单调性,由大到小
while (!que.empty() && que.back() < value) {
que.pop_back();
}
que.push_back(value);
}
void pop(int value) {
// 如果队列的首元素和value匹配,那么移除队列首元素
if (!que.empty() && que.front() == value) {
que.pop_front();
}
}
int front() {
return que.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> res;
myQueue que;
//对队列的初始化
for (int i = 0; i < k; i++) {
que.push(nums[i]);
}
//加入前k个的最大值
res.push_back(que.front());
int i = k;
while (i < nums.size()) {
que.pop(nums[i]);
que.push(nums[i]);
res.push_back(que.front());
i++;
}
return res;
}
};
/* main.cpp */
#include "myStack.h"
//滑动窗口的最大值
int main() {
int k = 3;
vector<int> nums{ 1,3,-1,-3,5,3,6,7 };
MyWindow solution;
vector<int> res;
res = solution.maxSlidingWindow(nums, k);
for (auto i : res) {
cout << i << " ";
}
}
算法效率
时间复杂度O(n)
空间复杂度O(k)
执行用时: 196 ms
内存消耗: 128.9 MB