PS:如果读过题了可以跳过题目描述直接到题解部分
提交链接:洛谷 P1886 滑动窗口 /【模板】单调队列
前言
临近CSP复赛,为了巩固基础,我开始返回来做模板题了。
题目
题目描述
有一个长为 n n n 的序列 a a a,以及一个大小为 k k k 的窗口。现在这个从左边开始向右滑动,每次滑动一个单位,求出每次滑动后窗口中的最大值和最小值。
例如:
The array is [ 1 , 3 , − 1 , − 3 , 5 , 3 , 6 , 7 ] [1,3,-1,-3,5,3,6,7] [1,3,−1,−3,5,3,6,7], and k = 3 k = 3 k=3。
输入格式
输入一共有两行,第一行有两个正整数
n
,
k
n,k
n,k。
第二行
n
n
n 个整数,表示序列
a
a
a
输出格式
输出共两行,第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
样例 #1
样例输入 #1
8 3
1 3 -1 -3 5 3 6 7
样例输出 #1
-1 -3 -3 -3 3 3
3 3 5 5 6 7
提示
【数据范围】
对于
50
%
50\%
50% 的数据,
1
≤
n
≤
1
0
5
1 \le n \le 10^5
1≤n≤105;
对于
100
%
100\%
100% 的数据,
1
≤
k
≤
n
≤
1
0
6
1\le k \le n \le 10^6
1≤k≤n≤106,
a
i
∈
[
−
2
31
,
2
31
)
a_i \in [-2^{31},2^{31})
ai∈[−231,231)。
题解
以求最小值,维护单增的单调队列为例:
每次加入数之前先把距离超过要求的数先删掉( + + l ++l ++l)(为了完成这一步,我们可以用单调队列存数的下标而非数字本身),然后维护它的单调性(将大小比即将加入的数大的数删掉( − − r --r −−r)),再把数加入单调队列,最后输出队首即可。
代码实现
100pts
//洛谷 P1886 滑动窗口 /【模板】单调队列
#pragma GCC optimize(3)
#include<iostream>
#include<cstdio>
using namespace std;
int n,k,a[1000010];
int b[100010];
int l,r;
int main(){
register int i;
scanf("%d %d",&n,&k);
for(i=1;i<=n;++i){
scanf("%d",&a[i]);
}
l=1,r=0;
for(i=1;i<=k;++i){
while(l<=r&&i-b[l]>=k){
++l;
}
while(l<=r&&a[b[r]]>=a[i]){
--r;
}
b[++r]=i;
}
printf("%d",a[b[l]]);
for(;i<=n;++i){
while(l<=r&&i-b[l]>=k){
++l;
}
while(l<=r&&a[b[r]]>=a[i]){
--r;
}
b[++r]=i;
printf(" %d",a[b[l]]);
}
printf("\n");
l=1,r=0;
for(i=1;i<=k;++i){
while(l<=r&&i-b[l]>=k){
++l;
}
while(l<=r&&a[b[r]]<=a[i]){
--r;
}
b[++r]=i;
}
printf("%d",a[b[l]]);
for(;i<=n;++i){
while(l<=r&&i-b[l]>=k){
++l;
}
while(l<=r&&a[b[r]]<=a[i]){
--r;
}
b[++r]=i;
printf(" %d",a[b[l]]);
}
printf("\n");
return 0;
}