牛客网:滑动窗口的最大值

这题是单调栈的典型应用。

我们维护一个值递减的下标序列。我们用一个双向队列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;
    }
};
经验分享 程序员 微信小程序 职场和发展