题目描述
给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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 k)
{
vector<int>a;
int len=num.size();
if(k>len||len==0)return a;
int max;
for(int i=0;i+k-1<len;i++){
max=num[i];
for(int j=i;j<k+i;j++){
if(num[j]>max)
max=num[j];
}
a.push_back(max);
}
return a;
}
};
方法二、双端队列
- 如果新来的值比队列尾部的数小,那就追加到后面,因为它可能在前面的最大值划出窗口后成为最大值
- 如果新来的值比尾部的大,那就删掉尾部(因为有更大的在后面,所以它不会成为最大值,划出也是它先划出,不影响最大值),再追加到后面,循环下去直到小于
- 如果追加的值比的索引跟队列头部的值的索引超过窗口大小,那就删掉头部的值
- 其实这样每次队列的头都是最大的那个
代码:
class Solution {
public:
vector<int> maxInWindows(const vector<int>& num, unsigned int size)
{
vector<int>a;
int len=num.size();
deque<int>q;
for(int i=0;i<len;i++){
while(q.size()&&num[i]>num[q.back()])
q.pop_back();
while(q.size()&&i-q.front()+1>size)
q.pop_front();
q.push_back(i);
if(size&&i>=size-1)
a.push_back(num[q.front()]);
}
return a;
}
};