洛谷 P1886 滑动窗口 /【模板】单调队列

PS：如果读过题了可以跳过题目描述直接到题解部分
提交链接：洛谷 P1886 滑动窗口 /【模板】单调队列

前言

临近CSP复赛，为了巩固基础，我开始返回来做模板题了。

题目

题目描述

有一个长为 $n$ 的序列 $a$，以及一个大小为 $k$ 的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。

例如：

The array is $[1,3,-1,-3,5,3,6,7]$, and $k=3$。

输入格式

输入一共有两行，第一行有两个正整数 $n,k$。
第二行 $n$ 个整数，表示序列 $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\%$ 的数据，$1\le n\le 10^5$；
对于 $100\%$ 的数据，$1\le k\le n\le 10^6$，$a_i\in [-2^{31},2^{31})$。

题解

以求最小值，维护单增的单调队列为例：

每次加入数之前先把距离超过要求的数先删掉（$++l$）（为了完成这一步，我们可以用单调队列存数的下标而非数字本身），然后维护它的单调性（将大小比即将加入的数大的数删掉（$--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;
}