剑指offer - 64滑动窗口的最大值

题目描述

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{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]}。

思路有下面两种：

1. 一般的解法，遍历数组，以当前元素为窗口的开始，再向后遍历窗口大小，分别对比出每个窗口的最大值

2. 使用一个队列辅助，遍历数组，将每个元素依次从队尾插入队列， 插入时判断，

    如果队尾元素比当前元素小，就出列， 插入元素之后判断，如果队首元素已经不在当前窗口中，则出列

    则队首元素就是当前窗口的最大值， 队列保存数组元素的下标

代码如下：

class Solution {
public:
    //方法一: 一般的解法，遍历数组，以每一个元素为窗口的开始，分别计算窗口的最大值
    vector<int> maxinwindows_common(const vector<int>& num, unsigned int size){
        vector<int> result;
        if(num.size() < size || 0 == size){
            return result;
        }
        for(int i = 0; i <= num.size() - size; i++){
            int max = num[i];
            for(int j = i+1; j <= i+size-1; j++){
                max = max < num[j] ? num[j] : max;
            }
            result.push_back(max);
        }
        return result;
    }
    //方法二：利用一个双端队列 保存数组元素的下标 数组元素依次从最后入队
    //元素从队列最后入队前，检测队尾元素如果小于当前值，则出列，即从队列的最后，小于当前值的出列
    //队首元素就是当前窗口的最大值
    //校验队首元素，是否已经不再当前窗口中，不在则出列
    vector<int> maxinwindows_dequeue(const vector<int>& num, unsigned int size){
        vector<int> result;
        deque<int> q;
        
        for(int i = 0; i < num.size(); i++){
            while(!q.empty() && num[q.back()] <= num[i]){ //从队尾校验，小于当前值的出列
                q.pop_back();
            }
            while(!q.empty() && i- q.front()+1 > size){   //从队首校验，已经不在当前窗口中的，出列
                q.pop_front();
            }
            q.push_back(i);
            if(size && i+1>=size){
                result.push_back(num[q.front()]);
            }
        }
        return result;
    }
    
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        //return maxinwindows_common(num, size);
        return maxinwindows_dequeue(num, size);
    }
};