题目(leecode T239):
给你一个整数数组 nums
,有一个大小为 k
的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k
个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
方法:滑动窗口,一进一出,我们可以使用队列的结构来解决。不过同时我们需要满足找到其中的最大值元素,理论上我们可以对队列内的K个元素进行比较找出最大值,不过这样的时间复杂度很高,而且通过分析我们可以发现我们没有必要将所有的元素都进行维护,而是只需要对潜在的最大值元素进行维护即可,同时我们可以将队列内的元素由大到小排列,由此我们可以自定义一个队列结构,其中的方法如下:
push方法:入栈方法,当栈为空时我们直接入栈,由于我们想让队列内的元素呈现由大到小的顺序,那么最大的元素应该在队列的头部,因此入栈时我们可以分析当前元素与栈尾元素的大小关系,如果当前元素大于栈尾元素,我们就将栈尾元素移出,将当前元素插入,否则我们就直接插入队尾即可。
pop方法:出栈方法,当栈不为空时可以出栈,判断当前元素与栈首元素是否相等,如果相等的话说明当前元素该出栈了,直接将元素出栈即可。但如果不相等的话沃恩不需要执行任何操作,这说明该出栈的元素现在已经不在栈内了。
front方法:该方法取出当前队列的最大值元素,因为我们一直维护的是一个单调递减的单调队列,当前队列元素的最大值,其实就是队首的元素。
题解:
class Solution {
private:
class MyQueue{
public:
deque<int> que;
void pop(int value){
if(!que.empty() && value == que.front()){
que.pop_front();
}
}
void push(int value){
while(!que.empty() && value > que.back()){
que.pop_back();
}
que.push_back(value);
}
int front(){
return que.front(); //单调队列的最大值就在队首
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
vector<int> result;
for(int i = 0; i < k; i++){ //先将前K个元素入队
que.push(nums[i]);
}
result.push_back(que.front()); //计算前K个元素的最大值
for(int i = k; i < nums.size(); i++){ //遍历剩下的元素
que.pop(nums[i - k]);
que.push(nums[i]);
result.push_back(que.front());
}
return result;
}
};