84. 柱状图中最大的矩形
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。
每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
这个问题和接雨水那道很相似,第一反应是使用单调栈做。
怎么确定当前矩形最大面积呢?
我们定义一个单调递增的栈,当新来的元素小于栈顶元素,那么栈顶元素对应的矩形面积是可以确定的,因为栈顶大于其左右两边的元素(这里我们假设右边元素是新加的元素)
计算完面积后,弹出栈顶元素,若新栈顶仍然大于新元素,那么重复上面的步骤,我们也可以确定新栈顶对应的最大面积
from typing import *
class Solution:
def largestRectangleArea(self, heights: List[int]) -> int:
"""
单调栈
首尾加两个哨兵,0
放在头的哨兵是防止栈为空
栈里面存放的是索引
:param heights:
:return:
"""
ans = 0
stack = [0]
hs = [0, *heights, 0]
for i, h in enumerate(hs):
while hs[stack[-1]] > h:
ans = max(ans, (i - stack[-2] - 1)*hs[stack.pop()])
stack.append(i)
return ans
S = Solution()
a = S.largestRectangleArea([2,1,5,6,2,3])
print(a)