这题是单调栈的典型应用。
我们维护一个值递减的下标序列。我们用一个双向队列deque来写,至于为什么,可以从后面的操作看出。
操作思路如下:
1.队列空,将下标压入队列
2.当前值比队列尾要大,将尾部弹出,直到队列空或者尾部下标对应的值比当前值大
3.压入当前值的下标
4.如果队列首部的元素下标加上size小于等于当前下标,说明之前的一个值不在滑动窗口内(由于每次只移动一个元素,所以只涉及到一个值)
5.如果当前下标已经大于size,说明窗口已经形成,将队列头部放入答案中
这样的时间复杂度就是O(n),空间复杂度也是O(n),比起暴力锁定一个窗口去遍历还是好很多的,暴力的话是O(size*n)
代码如下所示:
class Solution {
public:
vector<int> maxInWindows(const vector<int>& nums, int size) {
vector<int> res;
if(nums.size()==0 || size<1 || nums.size()<size){
return res;
}
int n=nums.size();
deque<int> dq;
for(int i=0;i<n;++i){
while(!dq.empty() && nums[dq.back()]<=nums[i]){
dq.pop_back();
}
dq.push_back(i);
if(dq.front()+size <= i){
dq.pop_front();
}
if(i+1>=size){
res.push_back(nums[dq.front()]);
}
}
return res;
}
};