牛客上看到这样一道题:
题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
解题思路:
看到这种滑动窗口, 有进有出的题目,很容易的想到应该考察的是队列的应用,而这道题很明显要用到双向队列。因为题目要求的是每个滑动窗口中的最大值,难点在于如何保存最大值,避免重复查找最大值,有一点动态规划的意思。 我们可以在队列头部(右边)保存最大值的index,每次读入一个数,保证两个条件:1、队列左边的index不过期,即在窗口内 2、队列右边保存的是窗口的最大值
下面是代码:
class Solution
{
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> ans;
unsigned int len = num.size();
if(len>=size && size>=1)
{
deque<int> Q;
for(unsigned int i = 0; i<size; ++i)
{
while(!Q.empty() && num[Q.back()]<=num[i])
Q.pop_back();
Q.push_back(i);
}
for(unsigned int i = size; i<len; ++i)
{
ans.push_back(num[Q.front()]);
while(!Q.empty() && num[Q.back()]<=num[i])
Q.pop_back();
if(!Q.empty() && Q.front()<=(int)(i-size))
Q.pop_front();
Q.push_back(i);
}
ans.push_back(num[Q.front()]);
}
return ans;
}
};