【单调队列】【模板】滑动窗口

链接

YbtOJ

题目描述

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

样例输入

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

样例输出

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

思路

单调队列维护最值模板

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

int n, k;
int a[1000006], q[1000005], p[1000005]; 

void minn()
{
	int head = 1, tail = 0;
	for(int i = 1; i <= n; ++i)
	{
		while(head <= tail && q[tail] >= a[i])
			tail--;
		q[++tail] = a[i];
		p[tail] = i;
		while(head <= tail && p[head] <= i - k)
			head++;
		if(i >= k)
			printf("%d ", q[head]);
	}
}

void maxx()
{
	int head = 1, tail = 0;
	for(int i = 1; i <= n; ++i)
	{
		while(head <= tail && q[tail] <= a[i])
			tail--;
		q[++tail] = a[i];
		p[tail] = i;
		while(head <= tail && p[head] <= i - k)
			head++;
		if(i >= k)
			printf("%d ", q[head]);
	}
}

int main()
{
	scanf("%d%d", &n, &k);
	for(int i = 1; i <= n; ++i)
		scanf("%d", &a[i]);
	minn();
	memset(q, 0, sizeof(q));
	memset(p, 0, sizeof(p));
	printf("\n");
	maxx();
	return 0;
}