原题来自牛客网剑指offer系列:链接
题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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^size)。
也可以采用维护双端单调队列的方法,时间复杂度O(n)。对新来的元素k,将其与双端队列中的元素相比较:
(1)后面比k小的,直接移出队列(因为不再可能成为后面滑动窗口的最大值了!)
(2)前面比k大的X,比较两者下标,判断X是否已不在窗口之内,不在了,直接移出队列
该单调队列保存下标,其中头元素为最大值的下标,且队列中不含不在窗口内的下标。
最终队列的第一个元素是滑动窗口中的最大值。
代码
(1)蛮力法:O(n*size)
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> ans;
if(size==0)
return ans;
for(int i=0;i<=(int)num.size()-(int)size;i++){
int max=num[i];
for(int j=i;j<i+size;j++){
if(num[j]>max)
max=num[j];
}
ans.push_back(max);
}
return ans;
}
};
(2)单调队列:O(n)
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])//从后面依次弹出队列中比当前num值小的元素,同时也能保证队列首元素为当前窗口最大值下标
s.pop_back();
while(s.size() && i-s.front()+1>size)//当当前窗口移出队首元素所在的位置,即队首元素坐标对应的num不在窗口中,需要弹出
s.pop_front();
s.push_back(i);//把每次滑动的num下标加入队列
if(size&&i+1>=size)//当滑动窗口首地址i大于等于size时才开始写入窗口最大值
res.push_back(num[s.front()]);
}
return res;
}
};
2021.3.4:
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
deque<int> s;
int len=num.size();
if(size>len)
return res;
for(int i=0;i<len;i++){
while(s.size()&&num[s.back()]<=num[i])
s.pop_back();
while(s.size()&&s.front()+size<=i)
s.pop_front();
s.push_back(i);
if(i>=size-1)
res.push_back(num[s.front()]);
}
return res;
}
};