题目:
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
思路:
使用双向队列,保存当前队列中的最大值位置,如果新滑入的值大于当前队列尾部的元素,则弹出队列尾部所有小于新值的元素,并把新滑入的值推入队列尾部;并判断队列头部保存的最大值的位置是否还在滑窗内,如果不在,则弹出该值;此时队列头部的值即为当前窗口最大值
代码实现:
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int> result;
int len = num.size();
if(len >= size && size > 0)
{
deque<int> max_idx;
for(int i = 0; i < size; i++)
{
while(!max_idx.empty() && num[i] >= num[max_idx.back()])
max_idx.pop_back(); // 弹出队列中小于当前值的所有数
max_idx.push_back(i);
}
for(int i = size; i < len; i++)
{
result.push_back(num[max_idx.front()]);
while(!max_idx.empty() && num[i] >= num[max_idx.back()])
max_idx.pop_back();
if(!max_idx.empty() && max_idx.front() <= i - size)
max_idx.pop_front(); // 保证队列中的最大位置在窗口内
max_idx.push_back(i);
}
result.push_back(num[max_idx.front()]);
}
return result;
}
};