单调队列与栈

单调队列与栈

单调队列

154. 滑动窗口 - AcWing题库

题目描述

给定一个大小为 $n\le 10^6$ 的数组。

有一个大小为 k 的滑动窗口，它从数组的最左边移动到最右边。

你只能在窗口中看到 k 个数字。

每次滑动窗口向右移动一个位置。

以下是一个例子：

该数组为 [1 3 -1 -3 5 3 6 7]，k 为 3。

窗口位置	最小值	最大值
[1 3 -1] -3 5 3 6 7	-1	3
1 [3 -1 -3] 5 3 6 7	-3	3
1 3 [-1 -3 5] 3 6 7	-3	5
1 3 -1 [-3 5 3] 6 7	-3	5
1 3 -1 -3 [5 3 6] 7	3	6
1 3 -1 -3 5 [3 6 7]	3	7

你的任务是确定滑动窗口位于每个位置时，窗口中的最大值和最小值。

输入格式

输入包含两行。

第一行包含两个整数 $n$ 和 $k$ ，分别代表数组长度和滑动窗口的长度。

第二行有 $n$ 个整数，代表数组的具体数值。

同行数据之间用空格隔开。

输出格式

输出包含两个。

第一行输出，从左至右，每个位置滑动窗口中的最小值。

第二行输出，从左至右，每个位置滑动窗口中的最大值。

输入样例：

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

输出样例：

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

朴素算法

每次向右 / 左移动就扫描一遍整个队列。那么时间复杂度为O(N*K)。显然不行。

单调队列做法

看名字就可以猜到，这个队列是单调的。比如说一个队列-3，3，-2，寻找最小值。如果3在。-2就一定在，而-2又比3小，那么这个3就可以出队了。总之我们就只要让队列是单调递增的，答案就是队首。

寻找最大值就是反着的。

之前写的有点草率。

先考虑最小值。什么时候一个数据没有价值呢？（价值哥

假设一个数在另一个数前面，前一个数为 a ，后一个为 b。因为是队列嘛，前面进后面出，所以说如果 a 在，b 一定在。这时假设 a 又大于 b ，我们求的是最小值，a 的存在就没有任何价值了是吧。那么我们在推入 b 之前判断一下 a 是否小于 b ，假如 a 小于 b ，那 a 就还有存在的价值，但如果 a 大于 b ，a 就可以滚了，他没有任何存在的价值。

举个例子更易懂一点。

假设窗口长度为 3。

• 将 1 入队
• 将 2 入队，发现 1 比 2 低，1 出队。1 此时已没有价值，如果再把 3 入队，最大值还是 2 ，之后 1 就出队了，所以没有任何价值。
• 将 3 入队，2 比 3 矮，留下来。这时最小值是 2 。2 就起到作用了吧。
• 再把 4 入队，没问题。
• 将 5 入队，此时 2 已出队，但是 3 4 都比 5 高，按顺序 3 4 分别出队。
• 将 6 入队，5 出队。

最大值亦是同理。反一下就好了。

代码

#include<iostream>
using namespace std;

#define N 1000010

int n,k,a[N];
int q[N];//存储下标

int main(){
	cin>>n>>k;
	for(int i=0;i<n;i++)
		cin>>a[i];
	int hh=0,tt=-1;//队头队尾
	for(int i=0;i<n;i++){
		if(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++){
		if(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;
	return 0;
}

单调栈

单调栈有几种不同种类。

法一

通过单调栈求前一个区间的最大或最小值。例如：

其实很简单嘛：就是求一直求最小值。最大值同理。

还可以想到的思路是反着求。虽然思路很简单，但是关键时刻可以提供帮助。

法二

通过栈求某一个数左边第一个比他小的数。

这看起来就高级很多吧。

思路也很简单，和上面单调队列差不多。同样是考虑价值。设 a 为数组中 b 前面的一个数。假设这个 a 大于 b ，a 就可以 出栈了。因为是求左边第一个小于该数的数，那么假如能用到 b 那怎么会用到 a，b 在 a 右边。栈里的数肯定是单调上升的。

再拿这个图做个示范

• 1 入栈没问题
• 2 入栈，1 出栈。
• 3 入栈没问题
• 4
• 5 比 3 4 都矮，所以把 3 4 出栈
• 6 是最矮的，就要把前面全部出栈。

代码

#include<iostream>
using namespace std;

#define N 100010

int n, tt, stack[N];

int main()
{
	cin >> n;
	for(int i = 1; i <= n; i ++)
	{
		int num;
		cin >> num;
		while(tt > 0 && stack[tt] >= num)
			tt --;
		if(tt != 0)
			cout << stack[tt] << " ";
		else
			cout << "-1" << " ";
		stack[++ tt] = num;
	}

	
	return 0;	
} 

2022-6-5 20:46:30