牛客上看到這樣一道題:
題目描述
給定一個數組和滑動窗口的大小,找出所有滑動窗口裏數值的最大值。例如,如果輸入數組{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;
}
};