[單調棧]模板題

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;
}