单调栈算法解释及应用/c++/

本博客来自南昌理工学院ACM集训队

算法概述

单调栈是单调的 就是具有单调性的栈结构，需要维护栈内元素递增或递减。
栈内元素单调递增的就是单调递增栈，栈内元素单调递减的就是单调递减栈。
单调栈中可以保存数组的下标或元素的值。

应用场景

本部分参考自单调栈·从零开始学算法.

• 可以在O(N)的时间复杂度，找出每个数左右两边第一个大于或小于它的解
• 单调递增栈用于查找两边第一个小于当前元素的值
• 单调递减栈用于查找两边第一个大于当前元素的值
• 一般数组中的单调性问题，题目中隐含第一个或离此元素最近的大于或小于元素的值，这类问题都可以考虑下，用单调栈是否可以求解

例题分析

例题

luogu p5788【模板】单调栈.
给你一个数组，输出每个数右边第一个比它更大的数的下标，如果没有则输出0

输入：
n=5
【1 4 2 3 5】
输出：
【2 5 4 5 0】

算法分析

暴力做法很好想也很好写，就是从头往后找，用双重for循环找到下一个比当前值大的元素。但这可是O(n^2)的复杂度，数据一大就被T飞。
而单调栈，可以将嵌套的for循环优化为一层，在线性时间内解决。

我们可以把问题转换一下，n个小朋友排队，问你每个小朋友向右看的第一个比他高的小朋友排在第几个。

对于任意一个小盆友X，都会遮住他后面所有比他矮的小小小盆友，那这个小盆友X和X之前的小盆友都不会看到他们，这些小小小盆友就可以溜了。

就比如说图中的2号小盆友，身高为4，挡住了他身后的3号和4号小盆友。
3号4号小盆友就可以快乐的去领盒饭了，此时2号小盆友看到的就是还没的5号小盆友。
综上所述：

• 应当选择从后向前扫

• 如果新加入的小盆友身高大于栈顶这个小盆友的身高，栈顶小盆友出栈溜啦溜啦

• 当前的答案就是栈顶元素，栈空则说明身后没有比他高的，答案为0

• 加入当前这个小盆友的位置到栈顶

代码及注释

int a[m];   //小盆友身高
int f[m];   //答案数组
int s[m];   //模拟栈
int n,x;
int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i++)
        scanf("%d", &a[i]);
    int top = 0;   //模拟栈顶
    for (int i = n; i >= 1; i--)
    {
        while(top>0&&a[i]>=a[s[top]])  //当前小盆友如果高于栈顶小盆友
            top--;    //栈顶小盆友溜啦
        if(top!=0)
        f[i] = s[top];   //记录答案
        else f[i]=0;
        s[++top] = i;     //加入小盆友的位置
    }
    for (int i = 1; i <= n; i++)
        printf("%d ", f[i]);   //输出答案
    return 0;
}