单调栈

单调栈

最近在做一些算法题时用到了单调栈的思想，所以在这里写一下我对于这种栈的理解。

栈，是数据结构里一个十分基础的概念，其最大的特点就是“先进后出”。
而至于单调栈，顾名思义，就是栈里的元素满足单调递增或单调递减。而我们如何评判单调栈是单调递增还是单调递减呢？
我们一般将栈顶到栈底定义为评判方向：从栈顶到栈底递增，为单调递增栈；从栈顶到栈底递减，为单调递减栈。

单调递增栈：
【6， 5， 2】（栈顶）

单调递减栈：
【2， 4， 8】（栈顶）

---

维护单调栈

明白了单调栈的定义后，下面我们来说说怎么维护一个单调栈：
假设我们要维护一个单调递增栈：【6， 5， 3， 2， 1】（栈顶）。现在，我们要插入一个元素 4 ，以下是插入过程：

步骤1： 栈顶元素1 与 4 比较，栈顶元素小，故pop栈顶元素1，栈变为【6， 5， 3， 2】；
步骤2： 栈顶元素2 与 4 比较，栈顶元素小，故pop栈顶元素2，栈变为【6， 5， 3】；
步骤3： 栈顶元素3 与 4 比较，栈顶元素小，故pop栈顶元素3，栈变为【6， 5】；
步骤4： 栈顶元素5 与 4 比较，终于，插入元素小，故可以插入4，栈变为【6， 5， 4】；

以上便是一个单调递增栈的维护过程，单调递减栈的维护也是同理。下面我用Python实现了一个单调递增栈：

a = [3, 4, 2, 6, 4, 5, 2, 3]

stack = []

for i in range(len(a)):
    num = a[i]
    while stack and stack[-1] < num:  # 比较， 循环直到 栈为空 或 栈顶元素比num小
        stack.pop()
    stack.append(num)  # 完成上述循环才可插入
    print(stack)

输出结果如下，大家可以先自己想一下这次的维护过程，再和我的结果比较：

---

例题

至于什么时候需要用到单调栈思想来帮助解题，我目前还没有很好地总结出来，所以在这里只分享例题给大家体验一下单调栈的利用：

请根据每日 气温 列表，重新生成一个列表。对应位置的输出为：
要想观测到更高的气温，至少需要等待的天数。如果气温在这之后都不会升高，请在该位置用 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，
你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

如果直接使用暴力循环，程序的耗时会超时，而单调栈可以大大减少时间复杂度：

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        length = len(T)
        ans = [0] * length  # 结果
        stack = []  # 单调栈
        for i in range(length):
            temperature = T[i]
            while stack and temperature > T[stack[-1]]:  # 若新的温度大于栈底元素  （注意：这里会依次比较， 直到栈底元素大于新温度
                prev_index = stack.pop()
                ans[prev_index] = i - prev_index  # 计算天数
            stack.append(i)
        return ans

以上便是本人对单调栈的理解，但我也只是刚刚接触，难免有不少错误，欢迎大家一起指正讨论。