题目来自于洛谷
P1886 滑动窗口 /【模板】单调队列
题目描述
有一个长为 nn 的序列 aa,以及一个大小为 kk 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [1,3,-1,-3,5,3,6,7][1,3,−1,−3,5,3,6,7], and k = 3k=3。
输入格式
输入一共有两行,第一行有两个正整数 n,kn,k。 第二行 nn 个整数,表示序列 aa
输出格式
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
很经典的单调队列问题,最重要的是以下两点:
一. 一定要注意 队列里存的是元素下标,并不是元素,通过下标就可以很方便的得到元素的值,从而输出。
二.保持栈的单调性,从而保证队头始终是最大,最小值
步骤有三:
1.去头 因为有滑窗的存在 数字只能存放有限个,当数字个数超出时,弹出队头。
2.去尾 这步是为了保证队列单调性,当新入队元素比队尾元素小时,弹出原队尾,新元素入队。
3.输出。
#include <iostream>
using namespace std;
const int N=1e6+5;
int n,k;
int a[N],q[N]; //模拟队列
int main(){
cin>>n>>k;
int hh=0,tt=-1;//队头队尾
for(int i=0;i<n;i++) cin>>a[i];
for(int i=0;i<n;i++){
if(hh<=tt&&q[hh]<i-k+1) hh++; //去头
while(hh<=tt&&a[q[tt]]>=a[i]) tt--;
q[++tt]=i;//去尾
if(i>=k-1) cout<<a[q[hh]]<<" ";//输出
}
cout<<endl;
hh=0,tt=-1;
for(int i=0;i<n;i++){
if(hh<=tt&&q[hh]<i-k+1) hh++;
while(hh<=tt&&a[q[tt]]<=a[i]) tt--;
q[++tt]=i;
if(i>=k-1) cout<<a[q[hh]]<<" ";
}//同理
cout<<endl;
}