滑动窗口[Acwing]

题目描述

题目传送门

输入样例
8 3
1 3 -1 -3 5 3 6 7
输出样例
-1 -3 -3 -3 3 3
3 3 5 5 6 7

思路

单调队列
维护一个单调队列，使其单调递增（队首为最小值） 或 单调递减（队首为最大值），我们可以在O（1）的时间内求出给定区间的最大（小）值

具体流程（以求最小值为例）
前期准备：用a数组存放数据，用q数组模拟单调队列，且q数组存放的是下标
1.由于单调队列的队首元素为最小值的下标，因此，我们需要确保队首元素在当前判断区间内。如当前遍历到下标i，则整个区间的下标范围为[i - k + 1, i]，因此需要将不符合要求的元素删除
2.为了确保队列是单调递增的，在加入新元素之前，需要将所有大于新元素的值删掉
3.如果队列不空，则最小值即为队首元素对应的值

AC代码

#include <bits/stdc++.h>
using namespace std;
const int N = 1000010;
int a[N], q[N];
int main()
{
	int hh = 0, tt = -1;
	int n, k;
	cin >> n >> k;
	for(int i = 0; i < n; i ++) cin >> a[i];
	
	//求窗口内的最小值
	for(int i = 0; i < n; i ++)
	{
		while(hh <= tt && q[hh] < i - k + 1) hh ++;
		while(hh <= tt && a[q[tt]] >= a[i]) tt --;
		q[++ tt] = i;
		if(i >= k - 1) cout << a[q[hh]] << " "; 
	} 
	cout << endl;
	//求窗口内的最大值
	hh = 0, tt = -1;
	for(int i = 0; i < n; i ++)
	{
		while(hh <= tt && q[hh] < i - k + 1) hh ++;
		while(hh <= tt && a[q[tt]] <= a[i]) tt --;
		q[++ tt] = i;
		if(i >= k - 1) cout << a[q[hh]] << " "; 
	} 
	return 0;
}

欢迎大家批评指正！！！