[C++] LeetCode 84. 柱状图中最大的矩形

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/lv1224/article/details/79974175

题目

给出 n 个非负整数来表示柱状图的各个柱子的高度,每个柱子紧挨彼此,且宽度为 1 。
您的函数要能够求出该柱状图中,能勾勒出来的最大矩形的面积。
示例:
这里写图片描述
以上是柱状图的示例,其中每个柱子的宽度为 1,给定的各个高度为 [2,1,5,6,2,3]
这里写图片描述
图中阴影部分为最大矩形面积,其面积为 10 个单位。
所以,您的函数给定高度是 [2,1,5,6,2,3] 的话,则返回 10

题解

这道题考虑用一个栈来维持一个递增序列,栈中保存的是当前元素的索引。如果当前值比栈顶元素小,那么栈中元素就需要弹出,直到栈为空或者栈顶元素小于当前元素。这样的话可以保证栈中第i个元素对应的高度总是小于等于栈中第i-1个元素至当前欲插入元素对应的矩阵高度,这样当元素弹出时,以它为最低高度的最大矩阵面积就很好求出来了。高直接可以得出,而长度则为第i-1个元素对应的索引至当前欲插入元素的索引的差值。
示例的过程如下:stack<int> st
矩阵1,高度为2,入栈,栈中元素为 0
矩阵2,高度为1,弹出栈顶元素并计算当前最大面积,弹出后栈为空,当前元素索引为1,故长度为1,面积为1*2=2,当前元素入栈,栈中元素为 1
矩阵3,高度为5,直接入栈,栈中元素为 1,2
矩阵4,高度为6,直接入栈,栈中元素为1,2,3
矩阵5,高度为2,小于栈顶元素对应矩阵高度,弹出栈顶元素3,栈非空,栈顶元素为2,故以矩阵4为最低高度的面积最大为6*(4-2-1)=6,栈中元素为1,2。此时当前矩阵高度依然小于栈顶元素对应矩阵高度,继续弹出2,栈非空,栈顶元素为1,故长度为4-1-1=2,面积为2*5=10。入栈,栈中元素为1,4
矩阵6,高度为3,大于栈顶元素,入栈,栈中元素为1,4,5
此时元素遍历完,弹出栈中元素。
弹出栈顶元素5,此时栈顶元素为4,此时长度为6-4-1=1,面积为1*3=3
弹出栈顶元素4,此时栈顶元素为1,此时长度为6-1-1=4,面积为4*2=8
弹出栈顶元素1,此时栈为空,此时长度为6,面积为1*6=6
注意:元素遍历完之后,栈顶元素对应的高度是小于等于其后所有矩阵的高度的,所以计算长度的时候要以矩阵个数来计算。
代码:

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        int n=heights.size();
        stack<int> index;
        int area=0;
        for(int i=0;i<heights.size();i++){
            if(index.empty()||heights[index.top()]<heights[i]) index.push(i);
            else{
                while(!index.empty()&&heights[index.top()]>=heights[i]){
                    int tmp=index.top();
                    index.pop();
                    int length=0;
                    if(index.empty()) length=i;
                    else length=i-index.top()-1;
                    area=max(area,length*heights[tmp]);
                }
                index.push(i);
            }
        }
        while(!index.empty()){
            int tmp=index.top();
            index.pop();
            int length=0;
            if(index.empty()) length=n;
            else length=n-index.top()-1;
            area=max(area,length*heights[tmp]);
        }
        return area;
    }
};
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页