题目描述
给一个滑动窗口的大小为W,先给你一个数组大小为N。现在滑动窗口一直滑动一直到末尾,求其次向右滑动时的最大值。
分析
常规思路
每次遍历一遍当前窗口,保存其中最大值。然后继续把窗口向右移动,最终做出答案。时间复杂度 O(N*W)。
优化
用一个双端队列,用它保存过程中的最大值不再每次遍历窗口。
具体这样的:
当队列为空
1.数组的下标 i尾插进去 。
当队列不为空
1.队尾元素对应的数大于下标 i 对应的数,下标 i 入队尾。
2. 否则,当队尾元素小于等于下标i对应的值,把队尾元素pop出去。
判断完的处理
1.当前 i 的值大于等于 w-1了,每次都把队头元素存到结果中。因为当这个条件为true了,说明当前 i 是第一个窗口的最后一个位置,从此开始窗口开始右滑,每次都要记录当前窗口最大值。
2.当 i 值等于 对头元素+W,说明当前对头元素已经不在窗口中,要过去了,就把它front_pop就行。
分析优化
为什么这样做能完成呢?
因为我们保证了在W窗口内,对头元素一直是当前窗口的最大值。大家仔细思考下。
其实这里就用了一个小技巧,用双端队列构造了一个容器,这个容器会保存着当前数组的最大值,这个容器我们可以称为 qmax。
这个题,稍微变化下,滑动窗口的最小值,qmin ,大家可以思考下。
代码
#include<iostream>
#include<vector>
#include<deque>
using namespace std;
#if 0
int GetMax(vector<int>& v, int begin, int end)
{
int ret = 0;
for (int i = begin; i <= end; ++i)
{
ret = (v[i] > ret) ? v[i] : ret;
}
return ret;
}
void window(vector<int> &v, vector<int>&ret,int w)
{
int size = v.size();
for (int i = 0; i < size-2; ++i)
{
ret.push_back(GetMax(v, i, i + w-1));
}
}
#endif
void window(vector<int>&v, vector<int>&ret,int w)
{
int size = v.size();
deque<int> decon;
for (int i = 0; i < size; ++i)
{
while (!decon.empty() && v[decon.back()]<=v[i])
{
decon.pop_back();
}
decon.push_back(i);
if (decon.front() + w == i)
{
decon.pop_front();
}
if (i >= w - 1)
{
ret.push_back(v[decon.front()]);
}
}
}