Leetcode 84. 柱状图中最大的矩形【单调栈】

Leetcode 84. 柱状图中最大的矩形

困难题; 使用单调栈解决此类问题。

题目

给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。

求在该柱状图中，能够勾勒出来的矩形的最大面积。

示例1

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

示例2

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

提示

• 1 <= heights.length <=105
• 0 <= heights[i] <= 104

解法

暴力解法

如果是暴力解法的话，针对每个元素找寻以此为高的矩形的左右边界，找边界的过程则需要O(n),总共的时间复杂度为$O(n^2)$；空间复杂度为O(1);

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        size = len(heights)
        res = 0

        for i in range(size):
            left = i
            cur_height = heights[i]
            while left > 0 and heights[left - 1] >= cur_height:
                left -= 1

            right = i
            while right < size - 1 and heights[right + 1] >= cur_height:
                right += 1

            max_width = right - left + 1
            res = max(res, max_width * cur_height)
        return res

• 时间复杂度为$O(n^2)$；
• 空间复杂度为O(1);

单调栈

可以维护一个单调栈，以空间代价交换时间。
维护一个单调递增栈，栈中元素存储下标；遍历到i位置位置，stack里面的j元素所能组成的最大面积的宽度即为从j-1元素位置到栈顶元素位置；

单调栈的考虑主要从元素压入时的当前处理元素，旧栈顶，新栈顶入手；建立相应的信息公式。

在本题中，栈中元素时递增，对于元素x而言，可以确定它的左矩形边界[也就是左边元素(下标)]，右边界待定，等待弹出的时候确定。

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        res,stack = 0, []
        heights.append(0)
        for i,new in enumerate(heights):
            while stack and heights[stack[-1]]>new: #new不能作为stack最后一个元素的扩展时就处理可以处理的元素
                right = i-1 # right一定是i-1 也就是栈顶元素
                cur = stack.pop()
                left = stack[-1] if stack else -1
                # print(cur, res, (right-left)*heights[cur])
                res = max(res, (right-left)*heights[cur])
            stack.append(i)
        return res

• 时间复杂度 O(n)
• 空间复杂度 O(n)