单调栈原理与应用
单调栈(Monotonic Stack)是一种特殊的栈数据结构,其核心特性是栈中的元素始终保持单调递增或单调递减的顺序。它的典型应用场景通常涉及需要快速找到某个元素左侧或右侧第一个比它大(或小)的元素的问题。
单调递减栈
- 原理
依次遍历数组,如果出现栈顶元素小于当前元素,则循环弹出栈顶元素,直到遇栈顶元素大于当前元素,则将当前元素入栈, - 应用
用来快速计算第一个比当前元素大的元素值
用来当时当前遍历遇到的最大值
单调递增栈
依次遍历数组,如果出现栈顶元素大于当前元素,则循环弹出栈顶元素,直到遇到栈顶元素小于当前元素,则将当前元素入栈,可以用来快速计算第一个比当前元素小的元素值
以下是
单调栈的主要应用场景及示例:
1. 下一个更大/更小元素(Next Greater/Smaller Element)
问题描述:给定一个数组,为每个元素找到其右侧(或左侧)第一个比它大(或小)的元素。
示例:
- 题目:496. 下一个更大元素 I
- 解法:维护一个单调递减栈,遇到比栈顶大的元素时,栈顶元素的下一个更大元素即为当前元素。
def nextGreaterElements(nums):stack = []res = [-1] * len(nums)for i in range(len(nums)):while stack and nums[i] > nums[stack[-1]]:res[stack.pop()] = nums[i]stack.append(i)return res
2. 柱状图中的最大矩形(Largest Rectangle in Histogram)
问题描述:在柱状图中找到面积最大的矩形。
示例:
- 题目:84. 柱状图中的最大矩形
- 解法:维护单调递增栈,确定每个柱子左右第一个比它矮的柱子边界。
def largestRectangleArea(heights):stack = []heights = [0] + heights + [0] # 添加哨兵节点res = 0for i in range(len(heights)):while stack and heights[i] < heights[stack[-1]]:h = heights[stack.pop()]w = i - stack[-1] - 1res = max(res, h * w)stack.append(i)return res
3. 接雨水(Trapping Rain Water)
问题描述:计算柱状图中能接多少雨水。
示例:
- 题目:42. 接雨水
- 解法:单调递减栈,计算凹槽的宽度和高度。
def trap(height):stack = []res = 0for i in range(len(height)):while stack and height[i] > height[stack[-1]]:bottom = height[stack.pop()]if not stack: breakw = i - stack[-1] - 1h = min(height[i], height[stack[-1]]) - bottomres += w * hstack.append(i)return res
4. 滑动窗口最大值(Sliding Window Maximum)
问题描述:给定数组和窗口大小,返回每个窗口中的最大值。
示例:
- 题目:239. 滑动窗口最大值
- 解法:使用单调递减的双端队列(类似单调栈思想)。
def maxSlidingWindow(nums, k):from collections import dequeq = deque()res = []for i in range(len(nums)):while q and nums[i] > nums[q[-1]]:q.pop()q.append(i)if q[0] == i - k: # 移除超出窗口的元素q.popleft()if i >= k - 1:res.append(nums[q[0]])return res
5. 其他变种问题
- 股票跨度问题(901. 股票价格跨度):维护单调递减栈,计算连续小于当前价格的天数。
- 子数组的最小值之和(907. 子数组的最小值之和):利用单调栈找到每个元素作为最小值的左右边界。
单调栈的核心思想
- 单调性选择:根据问题决定递增栈(解决“下一个更小”问题)或递减栈(解决“下一个更大”问题)。
- 出栈时机:当新元素破坏单调性时,出栈并处理结果(如计算面积、距离等)。
- 边界处理:通常通过添加哨兵节点(如柱状图问题中的首尾0)简化代码逻辑。
通过维护单调性,单调栈能将时间复杂度优化到 O(n),是处理“邻近比较”类问题的高效工具。
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.dtcms.com/a/257173.html
如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!