P1886 滑动窗口（洛谷）

P1886 滑动窗口（洛谷）

题目描述

有一个长为 $n$ 的序列 $a$，以及一个大小为 $k$ 的窗口。现在这个从左边开始向右滑动，每次滑动一个单位，求出每次滑动后窗口中的最大值和最小值。
例如，对于序列 $[1,3,-1,-3,5,3,6,7]$ 以及 $k=3$，有如下过程：
$$\begin{array}{ccc}\text{窗口位置}& \text{最小值}& \text{最大值}\\ \text{[1 3 -1] -3 5 3 6 7 }& -1& 3\\ \text{ 1 [3 -1 -3] 5 3 6 7 }& -3& 3\\ \text{ 1 3 [-1 -3 5] 3 6 7 }& -3& 5\\ \text{ 1 3 -1 [-3 5 3] 6 7 }& -3& 5\\ \text{ 1 3 -1 -3 [5 3 6] 7 }& 3& 6\\ \text{ 1 3 -1 -3 5 [3 6 7]}& 3& 7\end{array}$$

输入

输入一共有两行，第一行有两个正整数 $n,k$。
第二行 $n$ 个整数，表示序列 $a$

输出

输出共两行，第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值

样例输入

8 3
1 3 -1 -3 5 3 6 7

样例输出

-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})$。

题解1（C++版本）

#include<bits/stdc++.h>
using namespace std;

const int N = 1e6 + 10;

int n, k, a[N], mi[N], mx[N], q1[N], q2[N];

void getMax(){ //队头保存最大值 
	int h1 = 1, t1 = 0; //清空队列
	for(int i = 1; i <= n; i++){
		while(h1 <= t1 && a[q1[t1]] <= a[i]) t1--; //队尾出队 
		q1[++t1] = i; //队尾入队 
		if(q1[h1] < i - k + 1) h1++; //队头出队 
		if(i >= k) { // 使用最值 
			printf("%d", a[q1[h1]]);
			if(i < n) printf(" ");
		}
	} 
	printf("\n");
}

void getMin(){ //队头保存最小值 
	int h2 = 1, t2 = 0; //清空队列
	for(int i = 1; i <= n; i++){
		while(h2 <= t2 && a[q2[t2]] >= a[i]) t2--; //队尾出队 
		q2[++t2] = i; //队尾入队 
		if(q2[h2] < i - k + 1) h2++; //队头出队 
		if(i >= k) { // 使用最值 
			printf("%d", a[q2[h2]]);
			if(i < n) printf(" ");
		}
	} 
	printf("\n");
}
int main(){
	scanf("%d%d", &n, &k);
	for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
	getMin();getMax();
	return 0;
}