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

题目链接

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

题目描述

给定 $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$

解法 : 单调栈

我们维护一个单调递增的单调栈 $stk$。

如果此时 $heights[stk.top()]>heights[i]$，那么我们就需要更新答案，然后去掉栈顶元素，直到栈顶元素 $heights[stk.top()]\le heights[i[$ 即可。

红色的都是在栈 $stk$ 中的元素，当前 $heights[i]=3$。

接下来按照以下的三步，不断更新答案 $ans$ 即可：

• 记录栈顶元素高度 $h=heights[stk.top()]$；
• 弹出栈顶元素；
• 更新答案 ， a n s = m a x { a n s , h × ( i − s t k . t o p ( ) − 1 ) } ans = max \{ ans , h \times (i - stk.top() - 1) \} ans=max{ans,h×(i−stk.top()−1)}；

…
这样更新完即可。

特殊情况：对于只有一个元素的情况

我们可以在其头尾都插入 $0$ ，包含这种情况。

在这种情况下，也可以计算出单个矩形的面积。

时间复杂度： $O(n)$

C++代码：

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        //头尾插入 0
        heights.insert(heights.begin(),0);
        heights.emplace_back(0);

        int n = heights.size() , ans = 0;
        stack<int> stk;

        for(int i = 0;i < n;i++){
            while(!stk.empty() && heights[stk.top()] > heights[i]){
                int h = heights[stk.top()];
                stk.pop();
                //更新最大面积
                ans = max(ans , h * (i - stk.top() - 1));
            }
            stk.push(i);
        }

        return ans;
    }
};