题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路:用一个双向队列存下标 队列头存放最大的数的下标 ,新加入的数的下标 依靠数是否大于尾部判断 同时用下标差值表示窗口范围
拓展题目:实现一个队列的max函数 要求其push_back() pop_front()和函数max都是O(1)时间
思路: 成员变量包括数和下标 max函数对一个max双向队列进行队头操作获得最大值 其余情况下尾插和头出都和题目1方案一样
新增一个双向队列用于实现真正数据的push_back() pop_front() 其中当两者某数下标一致时 都出
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> res;
if(num.empty()||size<1||size>num.size())
return res;
deque<int> max;
for(int i=0;i<size;i++){
if(!max.empty()&&num[i]>=num[max.back()])
max.pop_back();
max.push_back(i);
}
res.push_back(num[max.front()]);
for(int i=size;i<num.size();i++){
if(!max.empty() && (i == max.front()+size))
max.pop_front();
while(!max.empty() && num[i] >= num[max.back()])
max.pop_back();
max.push_back(i);
res.push_back(num[max.front()]);
}
return res;
}
};