题目描述
这个思路也和单调队列的思路是相似的,就是要记录一些值使得下一次求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;
}