LeetCode 热题 HOT 100 第三十三天 84. 柱状图中最大的矩形 困难题 用python3求解

题目地址
给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。
求在该柱状图中，能够勾勒出来的矩形的最大面积。

示例 1
输入：heights = [2,1,5,6,2,3]
输出：10
解释：最大的矩形为图中红色区域，面积为 10

示例 2：

输入： heights = [2,4]
输出： 4

提示：
1 <= heights.length <=10^5
0 <= heights[i] <= 10^4

相似题：LeetCode 热题 HOT 100 第十九天 42. 接雨水

解法：单调栈

• 单调递增栈：单调递增栈就是从栈底到栈顶数据是从大到小，每当遇到比栈顶大的元素就弹栈
• 单调递减栈：单调递减栈就是从栈底到栈顶数据是从小到大，每当遇到比栈顶小的元素就弹栈

单调栈特别适合解决那些，两头大小决定中间值的大小的题。
因为单调栈中，以单调增栈为例。
设栈顶元素为 b， 栈顶第二个元素为a，自然有 a < b （因为堆栈越往上越大）
这时候， 若c出现，且c小于b， 那么b的左右第一个比b小的两个元素就找到了，分别是a和c，b在中间最大。
这时候你可以处理b，并重新整理堆栈，使其保持递增。
若c大于b，那c入栈，继续循环就行了。
最后清理堆栈

这个题的关键点在于：
以B点为高的矩形的最大宽度为：从a到c， 其中a，c分别为B左边和右边第一个小于B的元素。
单调栈的特点在于：
当遇见大数的时候， 压入堆栈，等待之后处理。
当遇见小数c的时候，意味着大数b的右边界c已经确定了。
这时候开始pop， 而以被pop出来的值（b）为高度的矩形的左右边界需要被确定。
其右边界就是当前的小数。即为c。左边界是堆栈下一层的元素，因为下一层的元素一定比当前小。且是第一小的元素。这时候a也确定了。
则以被pop出来的数为高度的矩形是 (c - a - 1) * pop()， 这里pop() == b

这里细节需要注意的是，

栈底要垫上-1，表示栈底。
循环结束，要清理堆栈。此时所有栈中继续存放的元素的右边界c都是结尾len(height)-1

代码实现：
解法一：

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        stack = [-1]
        max_res = 0
        for i in range(len(heights)):
            while len(stack) > 1 and heights[i] <= heights[stack[-1]]:
                max_res = max(max_res, heights[stack.pop()] * (i - stack[-1] - 1))  
            stack.append(i)
        for i in range(len(stack)-1):
            max_res = max(max_res, heights[stack.pop()]*(len(heights)-1-stack[-1]))
        return max_res

解法二：

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        stack = []
        heights = [0] + heights + [0]
        res = 0
        for i in range(len(heights)):
            while stack and heights[stack[-1]] > heights[i]:
                tmp = stack.pop()
                res = max(res, (i - stack[-1] - 1) * heights[tmp])
            stack.append(i)
        return res

注：
这个解法的巧妙之处在于：在数组heights的首尾加了各添加一个0。
因为能解决以下问题：如果遍历之后，栈不为空，那么还需要一步：即弹出栈中所有元素，分别计算最大面积。然而当加了两个0以后，在结束后，栈一定为空！
因为0入栈的时候，会把前面所有的数全部弹出，如果0最后一个入栈，所有的数都会弹出
不加0的话，导致栈中会一直有有效元素没有弹干净，因此需要在首尾加0，使得原数列中的所有有效元素均被弹出，从而获取最大值