题目描述
有一个整型数组arr和一个大小为w的窗口从数组的最左边滑到最右边,窗口每次向右边滑一个位置,求每一种窗口状态下的最大值。(如果数组长度为n,窗口大小为w,则一共产生n-w+1个窗口的最大值)
输入描述:
第一行输入n和w,分别代表数组长度和窗口大小
第二行输入n个整数Xi,表示数组中的各个元素
输出描述:
输出一个长度为n-w+1的数组res,res[i]表示每一种窗口状态下的最大值
示例1
输入
8 3 4 3 5 4 3 3 6 7
输出
5 5 5 4 6 7
备注:
1≤w≤n≤1000000 −1000000≤Xi≤1000000
算法思路:
进队列的策略:
1、如果dq为空,则直接将下标放入队列
2、如果dq非空,则比较data[i]和队列中的队尾j对应的data[j]的大小。
若data[j]>data[i],那么表示data[i]有可能在接下来成为最大值,所以将i放入队列
若data[j]<=data[i],那么表示data[j]已经不可能是最大的值了,所以将data[j]抛出,继续进行比较。
出队列的策略:
如果此时队首的下标k( dq.front() )满足k==i-w,则表示当前的队首元素已经过期,所以可以将队首的元素抛出。
//使用双端队列维护窗口最大值的角标
#include<bits/stdc++.h>
using namespace std;
int main(){
int n,w;
cin>>n>>w;
vector<int> data(n);
for(int i=0;i<n;i++){
cin>>data[i];
}
deque<int> dq;
vector<int> res(n-w+1);
int index=0;
for(int i=0;i<n;i++){
while(!dq.empty() && data[dq.back()]<=data[i]){
dq.pop_back();
}
dq.push_back(i);
if(dq.front()==i-w){
dq.pop_front();
}
if(i>=w-1){ //从该位置处,res中开始不断存储当前窗口的最大值
res[index++]=data[dq.front()];
}
}
for(auto x : res){
cout<<x<<" ";
}
return 0;
}