剑指Offer(三)滑动窗口的最大值

题目:

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

思路:

正常情况下,最笨的办法就是进行遍历,然后在每一个区间内去找出一个最大值。而本题的提示是使用栈和队列,因此是希望在时间复杂度为O(n)把问题解决。这题可以使用队列,图解如下,







程序代码(包含测试用的主程序)

#include<iostream>
#include<vector>
#include<deque>
using namespace std;

class Solution {
public:
vector<int> maxInWindows(const vector<int>& num,unsigned int size)
{
    vector<int> res;
    deque<int> s;
    for(unsigned int i=0;i<num.size();++i)
    {
        while(s.size()&&num[s.back()]<=num[i])
            s.pop_back();                     //当队列非空且待压入的元素比当前队尾大的时候,把当前队尾弹出
        while(s.size()&&i-s.front()+1>size)
            s.pop_front();                    //当队首已经不在滑动窗格内的时候,把队首元素移除
        s.push_back(i);                       //否则就压入当前的下标
        if(size&&i+1>=size)
            res.push_back(num[s.front()]);    //当前下标大于size时再压入最大值
    }
    return res;
}

    void print(vector<int> num)
    {
        for(vector<int>::iterator it=num.begin();it!=num.end();++it)
        {
            cout<<*it<<endl;
        }
    }
};


int main()
{
    vector<int> num;
    num.push_back(2);
    num.push_back(3);
    num.push_back(4);
    num.push_back(2);
    num.push_back(6);
    num.push_back(2);
    num.push_back(5);
    num.push_back(1);
    Solution s;
    vector<int> num1=s.maxInWindows(num, 3);
    s.print(num1);
}

猜你喜欢

转载自blog.csdn.net/bubbler_726/article/details/80922548
今日推荐