滑动窗口的最大值

来源:互联网 发布:淘宝用不了怎么回事 编辑:程序博客网 时间:2024/05/30 23:37

题目描述

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

这道题给定了一个数组,还给了一个窗口大小k,让我们每次向右滑动一个数字,每次返回窗口内的数字的最大值,而且要求我们代码的时间复杂度为O(n)。提示我们要用双向队列deque来解题,并提示我们窗口中只留下有用的值,没用的全移除掉。大概思路是用双向队列保存数字的下标,遍历整个数组,如果此时队列的首元素小于等于i-size时,队首元素不在滑动窗口了,移除队首元素。然后比较队尾元素和将要进来的值,如果小的话就都移除,然后此时我们把队首元素加入结果中即可,参见代码如下:

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

0 0
原创粉丝点击