哈啰,欢迎大家收看我的频道,那么我们继续更新算法题。今天带来的算法题是——柱状图中最大的矩形。还算是一道比较难且复杂的题目吧。那么我们先来看题。
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
示例 1:
输入:heights = [2,1,5,6,2,3] 输出:10 解释:最大的矩形为图中红色区域,面积为 10
示例 2:
输入: heights = [2,4] 输出: 4
可以看到,这次的算法题,核心就是需要我们求一下整个列表元素所能组成图形的最大面积,那么我们又应该用什么样的思路来进行解题呢?下面是我所用的方法:
解答:这道题,我所用的方法是单调栈的方法来解。所谓单调栈,就是我们先提前设定一个栈里元素,然后我们下面的元素不断的往下,一个接一个的与栈里元素比较,如果比栈里元素大,即为单调增,此时我们可以发现,所围成的最大面积就是最大的高度乘最大的宽度,但是如果当下一个元素出现单调递减的情况时,我们需要注意,虽然高度降低,但是我们的宽度还在增加,不排除我们因此获得最大面积的可能,所以,一旦出现下一个元素比栈里元素小的情况时,我们需要做的是,将前面的所有元素弹出,计算出所围成的面积,然后与之前的最大面积通过max函数进行比较,以此类推,知道所有的元素全部试完,输出最大面积即可。那么,理论存在,下面开始实践!
首先,我们我们要先对面积与单调栈进行初始化,然后我们将第一个元素输入到单调栈里,是为了防止以后运行时,单调栈的选取越界。
然后开始我们的循环,for循环用来限定元素个数,while循环用来判断是否单调,若不单调,即出现比之前元素小的情况时,我们就将前面的元素弹出,来逐个计算最大的面积,随后与之前的最大面积进行比较,利用max函数来将最大的面积保留下来即可。
最后,重复之前的循环,直至所有的元素全部判断结束为止,确定返回值,输出最大面积即可。思路就到这里,下面是代码:
class Solution:
def largestRectangleArea(self, heights: List[int]) -> int:
s = 0
stk = [0]
heights = [0] + heights + [0]
for i in range(1, len(heights)):
while heights[i] < heights[stk[-1]]:
n = stk.pop()
h = heights[n]
w = i - stk[-1] - 1
s = max(s, h*w)
stk.append(i)
return s
运行结果:
这道题总的来说,比之前的算法题难上不少,而且引用了一个新的方法知识叫做——单调栈,算法里还有跟这个类似的题,日后如果有不懂的家人们,我会再出一期博客去讲解另一道题。那么,如果大家有更好的思路以及解法,欢迎大家一起讨论,我们共同进步。我们下期再见。