84. 柱状图中最大的矩形
原始题目链接:https://leetcode.cn/problems/largest-rectangle-in-histogram/
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
解题思路:
使用单调栈解决两头大中间小或者两头小中间大的问题。例如以b点为高的矩形的最大宽度为, 从a到c,其中a,c分别为b左边和右边第一个小于b的元素,即找到b的边界a和c,矩形的面积就是(c - a - 1) * b代表的值。
代码实现:
class Solution:
def largestRectangleArea(self, heights: List[int]) -> int:
# 单调递增的栈,存储的是heights的元素下标
# 寻找的就是栈中的下标对应的元素的左右边界值
# 边界是比栈中元素下标对应的元素左右两边第一个小的元素所构成矩形边界
stack = [-1]
max_area = float('-inf')
for i in range(len(heights)):
# 从栈顶开始计算,如果找到比栈中元素小的第一个右边的边界
# 就开始计算矩形面积,左边第一个小的已经压入了栈因为栈是单调递增的
while len(stack) > 1 and heights[i] <= heights[stack[-1]]:
max_area = max(max_area, heights[stack.pop()] * (i - stack[-1] - 1))
# 没发现比栈顶小的元素,继续入栈,入栈的元素是heights的元素下标
stack.append(i)
# 如果栈中还剩下元素,计算栈中元素的最大矩形面积,此时右边界是heights的最后一个元素下标+1,即len(heights)
# 栈底是-1,所以索引到len(stack) - 1
for i in range(len(stack) - 1):
max_area = max(max_area, heights[stack.pop()] * (len(heights) - stack[-1] - 1))
return max_area
参考文献:
https://leetcode.cn/problems/largest-rectangle-in-histogram/solution/4ye-pptjie-shi-wei-he-ji-ru-he-jiang-dan-diao-zhan/