题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路:左程云的程序员面试代码指南,第19页。
利用双端队列来实现窗口最大值的更新。
首先生成双端队列qmax,qmax中存放数组的下标。
遍历到num[i],qmax的放入规则为:
1. 如果qmax为空,直接把下标i放入qmax;
2.如果qmax不为空,则设qmax队尾存放的下标为j,然后判断:
1)如果num[j] > num[i],那么将 i 放入qmax,因为当qmax中的 j 弹出后,那么 i 就有可能是后面的子序列的最大值的下标;
2)如果num[j] <= num[i],那么弹出 j (这里要用while,一直弹出直到qmax为空或者 num[j] 大于num[i]了,才可以保证队头保存的元素一直是num的子序列的最大元素的下标),因为j已经不可能是后面的子序列的最大值的下标了,所以直接弹出。
3. 如果qmax的队头保存的下标等于i - w,那么说明qmax的对头保存的下标已过期,弹出对头下标。
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
int len = num.size();
if(len == 0 || size == 0 || size > len)
return res;
deque<int> qmax;
for(int i = 0; i < len; ++i)
{
while(!qmax.empty() && num[qmax.back()] <= num[i])
qmax.pop_back();
if(qmax.empty() || num[qmax.back()] > num[i])
qmax.push_back(i);
if(qmax.front() == i - size)
qmax.pop_front();
if(i >= size - 1)
res.push_back(num[qmax.front()]);
}
return res;
}
};