P5788 模板单调栈

Posted 2021-11-12 辉小歌

https://www.luogu.com.cn/problem/P5788

#include<bits/stdc++.h>
using namespace std;
const int N=1e6*3+10;
int a[N],n,ans[N];
stack<int>st;
int main(void)
{
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i];
	for(int i=n;i>=1;i--)
	{
		while(st.size()&&a[st.top()]<=a[i]) st.pop();
		if(st.size()&&a[st.top()]>a[i]) ans[i]=st.top();
		st.push(i);
	}
	for(int i=1;i<=n;i++) cout<<ans[i]<<" ";
	return 0;
}