【基础算法模板】栈与队列:单调队列、单调栈

模板:

// tt表示栈顶
int stk[N], tt = 0;

// 向栈顶插入一个数
stk[ ++ tt] = x;

// 从栈顶弹出一个数
tt -- ;

// 栈顶的值
stk[tt];

// 判断栈是否为空,如果 tt > 0,则表示不为空
if (tt > 0)
{

}

队列

普通队列

模板:

// hh 表示队头,tt表示队尾
int q[N], hh = 0, tt = -1;

// 向队尾插入一个数
q[ ++ tt] = x;

// 从队头弹出一个数
hh ++ ;

// 队头的值
q[hh];

// 判断队列是否为空,如果 hh <= tt,则表示不为空
if (hh <= tt)
{

}

循环队列

模板:

// hh 表示队头,tt表示队尾的后一个位置
int q[N], hh = 0, tt = 0;

// 向队尾插入一个数
q[tt ++ ] = x;
if (tt == N) tt = 0;

// 从队头弹出一个数
hh ++ ;
if (hh == N) hh = 0;

// 队头的值
q[hh];

// 判断队列是否为空,如果hh != tt,则表示不为空
if (hh != tt)
{

}

单调栈

作用:给定一个序列,找出每个数左边离它最近的比它大/小的数

模板:

常见模型:找出每个数左边离它最近的比它大/小的数
int tt = 0;
for (int i = 1; i <= n; i ++ )
{
    while (tt && check(stk[tt], i)) tt -- ;
    stk[ ++ tt] = i;
}

原理:

  • 遍历数组元素,每处理完一个元素就将它放入栈中,根据后进先出的特性,栈顶元素就是当前元素左边的最近元素,从栈顶开始往外弹出元素直到找到一个比当前元素小的元素停止。问题在于我们弹出的元素是否可能是接下来元素的答案?如果下一个元素大于当前元素,那么当前元素就是下一个元素的答案,与我们弹出的元素无关;如果下一个元素小于等于当前元素,因为弹出的元素都是大于等于当前元素的,所以弹出的元素依然不会是答案;说明弹出的元素不会影响后续结果对每个元素进行入栈或出栈操作只可能为一次,这也是为什么这个问题可以用单调栈解决。

测试:

/*
题目描述:
给定一个长度为N的整数数列,输出每个数左边第一个比它小的数,如果不存在则输出-1

输入格式:
第一行包含整数N,表示数列长度
第二行包含N个整数,表示整数数列

输出格式:
1<=N<=10e5
1<=元素值<=10e9

输入样例:
5
3 4 2 7 5

输出样例:
-1 3 -1 2 2 

*/ 


#include <iostream>
using namespace std;

const int N = 10e5 + 10;

int n;
int stk[N], tt;


int main(){
	scanf("%d", &n);
	
	for(int i = 0; i < n; i ++){
		int x;
		scanf("%d",&x);
		while(tt && stk[tt] >= x) tt --;
		if(tt) printf("%d ", stk[tt]);
		else printf("-1 ");
		
		stk[ ++ tt] = x;
	}
	
	return 0;
}

单调队列

作用:求滑动窗口中的最大值/最小值

模板:

常见模型:找出滑动窗口中的最大值/最小值
int hh = 0, tt = -1;
for (int i = 0; i < n; i ++ )
{
    while (hh <= tt && check_out(q[hh])) hh ++ ;  // 判断队头是否滑出窗口
    while (hh <= tt && check(q[tt], i)) tt -- ;
    q[ ++ tt] = i;
}

原理:

  • 以 求滑动窗口中的最小值 为例,
  • 窗口位置                    最小值  
    1 [3 -1 -3] 5 3 6 7           -3            
  • 当下队列出队了 1,入队了 -33-3 的左边,之后 3 肯定比 -3 先弹出去且 3-3 大,换言之只要 -3 在一天,3 就永远不可能作为最小值输出。-1 也是同理。所以只要有元素 在当前入队元素左边且比入队元素大,就把该元素删除,剩下的队列元素就是单调递增的。这个递增序列的第一个元素就是最小值。
  • 怎么知道队首元素,什么时候出队?q[ hh ] 队列存的是下标,不是元素值。我们知道当前元素的下标 i 与窗口长度 k ,判断队头下标 q[ hh ] 不在 i - k +1 ~ i 之间,则队头元素 a[ q[ hh ] ] 出队。
  • 求最大值同理。

测试:

/*
题目描述:
给定一个大小为 n<=10e6 的数组
有一个大小为 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 


*/

#include <iostream>
using namespace std;

const int N = 1e6 + 10;

int n, k;
int a[N], q[N];

int main(){
	scanf("%d%d", &n, &k);
	for(int i = 0; i < n; i ++)	scanf("%d", &a[i]);
	
	int hh = 0, tt = -1;
	for(int i = 0; i < n; i ++){
		//判断对头是否已经滑出窗口 
		if(hh <= tt && i - k + 1 > q[hh])	hh ++;
		while(hh <= tt && a[q[tt]] >= a[i])	tt --;
		q[ ++ tt] = i;
		if(i >= k - 1)	printf("%d ", a[q[hh]]);	
	}
	puts("");
	
	hh = 0, tt = -1;
	for(int i = 0; i < n; i ++){
		//判断对头是否已经滑出窗口 
		if(hh <= tt && i - k + 1 > q[hh])	hh++;
		while(hh <= tt && a[q[tt]] <= a[i])	tt --;
		q[ ++ tt] = i;
		if(i >= k - 1)	printf("%d ", a[q[hh]]);
	}
	puts("");
	
	return 0;
}

  • 8
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值