单调栈 (模板)2023/2/26

P5788 【模板】单调栈 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

#include <iostream>
#define int long long
using namespace std;
const int N = 3e6+10;
int stk[N],a[N],b[N];
int tt,n;
signed main()
{
	cin >> n;
	for(int i = 1;i <= n;i ++)
	{
		cin >> a[i];
	}
	for(int i = n;i >= 1;i --)
	{
		while(tt&&a[stk[tt]]<=a[i]) tt--;
		if(tt) b[i]=stk[tt];
		else b[i]=0;
		stk[++tt]=i;
	}
	for(int i = 1;i <= n;i ++)
	{
		printf("%lld ",b[i]);
	}
	return 0;
}

830. 单调栈 - AcWing题库

#include <iostream>
using namespace std;
int stk[10086];
int tt;
int main()
{
	int n;
	cin >> n;
	for(int i=1;i<=n;i++)
	{
		int x;
		cin >> x;
		while(tt&&stk[tt]>=x) tt--;
		if(tt) printf("%d ",stk[tt]);
		else printf("-1 ");
		stk[++tt]=x;
	}
	return 0;
}

单调栈是一种特殊的栈结构,它可以快速找到栈中的最大值或最小值。单调栈常用于解决一些数组、字符串中的问题,例如求下一个更大的元素、求前一个更大的元素等等。 下面是单调栈的代码模板: ``` stack<int> mono_stack; for (int i = 0; i < n; i++) { while (!mono_stack.empty() && mono_stack.top() > x[i]) { mono_stack.pop(); } mono_stack.push(x[i]); } ``` 其中,`x` 是一个数组,`n` 是数组的长度,`mono_stack` 是单调栈。这段代码的意思是:遍历数组 `x`,对于每个元素 `x[i]`,将其压入栈中,然后不断弹出栈顶元素,直到栈为空或者栈顶元素不大于 `x[i]`。这样,最终单调栈中的元素就是一个单调递增的序列。 如果要求下一个更大的元素,可以在遍历数组时,记录每个元素的下标,然后在弹出栈顶元素时,更新下标对应的答案。代码如下: ``` stack<pair<int, int>> mono_stack; for (int i = 0; i < n; i++) { while (!mono_stack.empty() && mono_stack.top().first < x[i]) { ans[mono_stack.top().second] = x[i]; mono_stack.pop(); } mono_stack.push(make_pair(x[i], i)); } ``` 其中,`ans` 是答案数组,`ans[i]` 表示第 `i` 个元素的下一个更大的元素。在弹出栈顶元素时,更新对应下标的答案即可。 如果要求前一个更大的元素,可以将数组逆序遍历,然后按照同样的方式求解。代码如下: ``` stack<pair<int, int>> mono_stack; for (int i = n - 1; i >= 0; i--) { while (!mono_stack.empty() && mono_stack.top().first < x[i]) { ans[mono_stack.top().second] = x[i]; mono_stack.pop(); } mono_stack.push(make_pair(x[i], i)); } ``` 以上就是单调栈的代码模板,可以根据具体问题进行修改和扩展。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值