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