滑动窗口的最大值

题目描述

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3，那么一共存在6个滑动窗口，他们的最大值分别为{4,4,6,6,6,5}； 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个： {[2,3,4],2,6,2,5,1}， {2,[3,4,2],6,2,5,1}， {2,3,[4,2,6],2,5,1}， {2,3,4,[2,6,2],5,1}， {2,3,4,2,[6,2,5],1}， {2,3,4,2,6,[2,5,1]}。

代码

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> ans;
        if (num.size() > 0 && size > 0 && size <= num.size()) {
            if (size > num.size()) {
            	size = num.size();
            }
            queue<int> maxQue;
            maxQue.push(0);
            
            for (int i = 1; i < size; ++i) {//0到size-1窗口的队列
                while (!maxQue.empty() && num[i] > num[maxQue.front()]) {//当前数大于队列里的数，队列要出队
                    maxQue.pop();
                }
                maxQue.push(i);
            }
            ans.push_back(num[maxQue.front()]);
            
            for (int i = size; i < num.size(); ++i) {
                if (i - maxQue.front() >= size) {//队列中的数超出了窗口
                    maxQue.pop();
                    while (!maxQue.empty() && num[i - 1] > num[maxQue.front()]) {//窗口可能为1，队列最后一个元素下标肯定是i-1
                        maxQue.pop();
                    }
                }
                while (!maxQue.empty() && num[i] > num[maxQue.front()]) {//当前数大于队列里的数，队列要出队
                    maxQue.pop();
                }
                maxQue.push(i);
                ans.push_back(num[maxQue.front()]);
            }
        }
        return ans;
    }
};