模板题：单调栈（洛谷p5788）

题目背景

模板题，无背景。

2019.12.12 更新数据，放宽时限，现在不再卡常了。

题目描述

给出项数为 nn 的整数数列 a1…na1…n​。

定义函数 f(i)f(i) 代表数列中第 i个元素之后第一个大于 a[i] 的元素的下标，即 f(i)=min⁡i<j≤n,aj>ai{j}f(i)=mini<j≤n,aj​>ai​​{j}。若不存在，则 f(i)=0f(i)=0。

试求出 f(1…n)f(1…n)。

输入格式

第一行一个正整数 nn。

第二行 nn 个正整数 a1…na1…n​。

输出格式

一行 nn 个整数表示 f(1),f(2),…,f(n)f(1),f(2),…,f(n) 的值。

输入输出样例

输入 

5
1 4 2 3 5

输出 

2 5 4 5 0

先看代码然后解释

#include<iostream>
#include<cstdio>

using namespace std;
const int N=3000010;
//sta为模拟栈的数组，a[n]存需要处理的元素，tt控制栈的大小 
int sta[N],a[N],tt=0,q[N]={0};//q[N]初始化为0，在倒序处理时，入栈元素为a数组最后一个，后面没有比这个数更大的，索引正好为0。 


int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    
    for(int i=n;i>0;i--)
    {
        while(tt&&a[i]>=a[sta[tt]]) tt--;//当栈顶元素小于a[i]时弹出栈顶元素 （栈底到栈顶为递减） 
		if(tt) q[i]=sta[tt];//当栈不为空时q[i]存入符合题意的索引值，这个索引值为栈顶元素（sta[tt]） 
		sta[++tt]=i;//把a[i]的下标i放到栈顶（压栈） 
    }
    for(int i=1;i<=n;i++)
    {
        printf("%d ",q[i]);//将结果正序打印 
    }
    return 0;
}

本题采用倒序处理，如题中数据1 4 2 3 5。

  当 i=5 时，a[5]=5 栈为空（tt初始值为0，tt>0时栈不为空），不执行while(tt&&a[i]>=a[sta[tt]]) tt--;  然后因为本右侧没有比它大的元素 所以  q[5]=0 然后将当前的元素的下标压入栈sta[1]=5（sta[++tt]=i）。

  当i=4时，a[4]=3,tt=1。栈不为空但a[4]>=a[sta[1]]不成立while结束，栈不为空所以q[4]=5  (q[i]=sta[tt],q[4]=5 中的5就是a[4]右侧第一个大于a[4]的元素的下标)，sta[2]=4。后面操作以此类推

可以看到最终q数组所存的值就是本题的答案。

本文为个人笔记，如有题目侵权，请联系删除。