188、【栈与队列】leetcode ——84. 柱状图中最大的矩形：暴力解法+单调栈（C++/Python版本）

题目描述

Problem: 84. 柱状图中最大的矩形

解法一：暴力解法

解题方法

采用暴力遍历方式，分别以每个柱子为高度，找到以柱子i为高度的合法矩阵区域。此时的矩阵区域的条件一定是左边界高度大于等于柱子i，右边界高度大于等于柱子i，每遇到柱子i大于左右边界的时候，就退出探查，计算结果。

C++

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        int res = 0, n = heights.size();
        for(int i = 0; i < n; i++) {
            int left, right;
            for(left = i; left >= 0; left--) {              // 向左探寻，找到左边界
                if(heights[left] < heights[i])
                    break;
            }
            for(right = i; right < n; right++) {            // 向右探寻，向右探寻找到右边界
                if(heights[i] > heights[right])
                    break;
            }
            int w = right - left - 1;                       // 取边界中间的面积
            int h = heights[i];
            res = max(res, w * h);
        }

        return res;
    }
};

此时时间复杂度为 $O(n^2)$

Python

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        res, n = 0, len(heights)
        for i in range(0, n):
            left, right = i, i
            for _ in range(left, -1, -1):                
                if heights[left] < heights[i]:
                    break
                left -= 1
            for _ in range(right, n):
                if heights[i] > heights[right]:
                    break
                right += 1                
            res = max(res, (right - left - 1) * heights[i])
        
        return res

解法二：单调栈

整体思路与 42. 接雨水 相近，区别在于接雨水的时候，单调栈是按从栈底到栈顶单调不增顺序存储，而本题要找到最大勾画面积，可以让每次到达一个比栈顶元素更低的长方体时，触发求面积的计算，也就是每当遇到右边第一个比自己更小的元素时，计算一次。因此，本体可按照单调栈从栈底到栈顶，单调不减的次序存储。

解题方法

采用单调栈的方式，自栈底到栈顶，按照单调不减的次序存储元素。每当第一次遇到比栈顶元素更小的数，则将栈顶元素弹栈，计算高度差和间隔差，得到面积，再和已有结果比较取最大值。

注意：由于有可能输入数组会是单调递增数组，那么如果按照原数组进行操作，就不能找到结果。因此，需要再数组末尾添加一个0，来保证可以触发计算。同时，为了防止第一个入栈元素就会被弹出，在后续还要进行求间隔差操作，为了避免空栈导致不能得到st.top()，因此再数组头需要再插入一个0元素，让第一个数就进行运算时可正常运算不需要单独再写代码，统一了操作。

复杂度

• 时间复杂度:

$O(n)$

• 空间复杂度:

$O(n)$

Code

C++

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        heights.insert(heights.begin(), 0);         // 首端加0，为了避免计算到只剩一个元素时，出现空栈现象，加一个0
        heights.push_back(0);                       // 尾端加0，保证升序序列也可以被正常计算
        int n = heights.size(), res = 0;
        stack<int> st;
        st.push(0);

        for(int i = 1; i < n; i++) {
            while(heights[st.top()] > heights[i]) {
                int mid = st.top();             st.pop();       // 获取左侧的位置下标并从栈中弹出
                int w = i - st.top() - 1;                       // 获取中间的这段长度
                int h = heights[mid];                           // 获取高度
                res = max(res, h * w);
            }
            st.push(i);
        }
        
        return res;
    }
};

Python

class Solution:
    def largestRectangleArea(self, heights: List[int]) -> int:
        heights.insert(0, 0)
        heights.append(0)
        st = [0]
        res, n = 0, len(heights)

        for i in range(1, n):
            while heights[st[-1]] > heights[i]:
                mid = st[-1]
                st.pop()
                w = i - st[-1] - 1
                h = heights[mid]
                res = max(res, h * w)
            st.append(i)
        
        return res

参考文章：84. 柱状图中最大的矩形