第十章 单调栈(三)
主要内容
单调栈
84. 柱状图中最大的矩形
思路分析
找到当前柱子左右第一个小于其高度的坐标,以其为宽,以当前高度为高,形成面积
前后插入0,头插防溢出,尾插为了把其他的都弹出
最后stack=[0,n]
代码
class Solution:
# 单调栈,从栈底到栈头是递增的,碰到右边第一个小于栈头的元素,栈头弹出,计算宽度,面积
def largestRectangleArea(self, heights: List[int]) -> int:
heights = [0] + heights + [0]
n = len(heights)
res = 0
stack = [0]
for i in range(1, n):
while stack and heights[i] < heights[stack[-1]]:
h = heights[stack.pop()]
w = i - stack[-1] - 1
res = max(res, h * w)
stack.append(i)
return res