剑指offer——滑动窗口的最大值（c++）

题目描述

给定数组，和滑动窗口的大小，请找出所有滑动窗口内的最大值。
例如，数组{2,3,4,2,6,2,5,1}，滑窗大小为3，那么结果为{4,4,6,6,6,5}。

思路

使用双向队列，用来保存有可能是滑动窗口最大值的数组的下标。在存入一个数字的下标之前，首先要判断队列里已有数字是否小于待存入的数字。如果已有的数字小于待存入的数字，那么这些数字已经不可能是滑动窗口的最大值，因此需要在队列尾部删除（pop_back）。同时，如果队列头部的数字已经从窗口里滑出，那么滑出的数字也需要从队列的头部删除（pop_front）。
数组{2,3,4,2,6,2,5,1}的长度为3的滑动窗口最大值求解步骤如下

步骤    插入数字    滑动窗口    队列中的下标   最大值
1       2          2           0(2)          N/A          
2       3          2,3         1(3)          N/A   
3       4          2,3,4       2(4)          4   
4       2          3,4,2       2(4),3(2)     4   
5       6          4,2,6       4(6)          6   
6       2          2,6,2       4(6),5(2)     6   
7       5          6,2,5       4(6),6(5)     6   
8       1          2,5,1       6(5),7(1)     5

代码

class Solution{
public:
    vector<int> maxInWindow(vector<int>& nums, unsigned int size){
        vector<int> res;
        if(nums.size() == 0 || size > nums.size() || size < 1)
            return res;
        deque<int> dq;
        for(int i = 0; i < size; ++i){
            while(!dq.empty() && nums[i] >= nums[dq.back()]){
                dq.pop_back();
            }
            dq.push_back(i);
        }
        for(int i = size; i < nums.size(); ++i){
            res.push_back(nums[dq.front()]);
            while(!dq.empty() && nums[i] >= nums[dq.back()])
                dq.pop_back();
            if(!dq.empty() && dq.front() <= (int)(i - size))
                dq.pop_front();
            dq.push_back(i);
        }
        res.push_back(nums[dq.front()]);
        return res;
    }
};