P1886 滑动窗口 【单调队列】

题目描述

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

例如：

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=3k = 3k=3。

输入格式

输入一共有两行，第一行有两个正整数 n,kn,kn,k。 第二行 nnn 个整数，表示序列 aaa
输出格式

输出共两行，第一行为每次窗口滑动的最小值
第二行为每次窗口滑动的最大值
输入输出样例
输入 #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≤1051 \le n \le 10^51≤n≤105；
对于 100%100%100% 的数据，1≤k≤n≤1061\le k \le n \le 10^61≤k≤n≤106，ai∈[−231,231)a_i \in [-2^{31},2{31})ai​∈[−231,231)。

鉴于上题的教训，单调队列还是手敲吧。

#include <iostream>
#include <algorithm>
#include <queue>
#include <cstdio>
using namespace std;

const int maxn = 1e6 + 10;
int a[maxn];
int id[maxn];
int data[maxn];
int head, tail;
int n, m;

void min_deque() {
	for(int i = 1; i <= n; i++) {
		while(head < tail && id[head] <= i - m)	head++;
		while(head < tail && data[tail - 1] >= a[i])	tail--;
		data[tail] = a[i];
		id[tail] = i;
		tail++;
		i >= m ? printf("%d ", data[head]) : printf("");
	}
}

void max_deque() {
	head = 0, tail = 0;
	for(int i= 1; i <= n; i++) {
		while(head < tail && id[head] <= i - m)	head++;
		while(head < tail && data[tail - 1] <= a[i])	tail--;
		data[tail] = a[i];
		id[tail] = i;
		tail++;
		i >= m ? printf("%d ", data[head]) : 0;
	}
}

int main() {
	cin >> n >> m;
	for(int i = 1; i <= n; i++)
		scanf("%d", &a[i]);
	min_deque();
	printf("\n");
	max_deque();
	return 0;
}