【算法】单调栈
一、单调栈
单调栈(Monotonic Stack)是一种特殊的栈结构,其核心思想是维护栈内元素的单调性(单调递增或单调递减)。单调栈通常用于解决与元素大小关系相关的问题,例如:
找到数组中每个元素的下一个更大(或更小)元素。
计算数组中每个元素左边或右边的第一个更大(或更小)元素。
解决与区间最值相关的问题(如最大矩形面积)。
二、特点
单调性:栈内元素始终保持单调递增或单调递减。
操作规则:
如果新元素满足单调性,直接压入栈。
时间复杂度:由于每个元素最多入栈和出栈一次,因此时间复杂度通常为O(n)
如果新元素破坏了单调性,则弹出栈顶元素,直到满足单调性为止。
三、应用场景
下一个更大元素:
给定一个数组,找到每个元素的下一个更大元素。每日温度:
给定一个温度数组,计算每天需要等待多少天才能遇到更高的温度。最大矩形面积:
给定一个柱状图,计算其中最大的矩形面积。滑动窗口最大值:
在滑动窗口中快速找到最大值。
四、步骤
以找到数组中每个元素的下一个更大元素为例:
初始化:创建一个空栈和一个结果数组。
遍历数组:
如果当前元素大于栈顶元素,则栈顶元素的下一个更大元素就是当前元素。
弹出栈顶元素,并更新结果数组。
重复上述过程,直到当前元素小于等于栈顶元素或栈为空。
将当前元素压入栈。
处理剩余元素:遍历结束后,栈中剩余的元素没有下一个更大元素,将其结果设置为
-1。