[单调栈]模板题

P5788 【模板】单调栈

题目描述

这个思路也和单调队列的思路是相似的，就是要记录一些值使得下一次求f(i)的时候不把它右边的数全部遍历一遍。从右往左看，如果新加入的这个数比之前的数字还大的话，要原来的数就没有什么用了，所以构造的单调栈是一个单调递减的栈。

AC代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll maxn=3e6+20;

ll n,a[maxn];
ll ans[maxn];
stack <ll> S;

int main(){
	scanf("%lld",&n);
	for(ll i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	}
	
	ans[n]=0;
	S.push(n);
	for(ll i=n-1;i>=1;i--){
		while(S.size()&&a[S.top()]<=a[i])
			S.pop();
			
		if(S.size())
			ans[i]=S.top();
		else
			ans[i]=0;
			
		S.push(i);
	}
	
	for(ll i=1;i<=n;i++){
		if(i!=n)
			printf("%lld ",ans[i]);
		else 
			printf("%lld",ans[i]);
	}
	return 0;
}