leetcode84.柱状图中最大的矩形

参考:
https://blog.csdn.net/lv1224/article/details/79974175

思路:单调栈

用一个栈来维持一个递减序列(从栈顶到栈底递减),栈中保存的是当前元素的索引。那么要弹出元素的高度大
于当前元素也大于栈中的倒数第二个元素
矩阵1,高度为2,入栈,栈中元素为 0
矩阵2,高度为1,弹出栈顶元素并计算以矩阵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
注意:元素遍历完之后,栈顶元素对应的高度是小于等于其后所有矩阵的高度的,所以计算长度的时候要以矩阵
个数来计算。

假设当前元素下标为cur,要弹出元素下标为p,那么以要弹出元素为最高点的矩形面积为(cur-p-1)*height[p]
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;
    }
};

为了使代码简单一些,去掉最后的那个while循环可以考虑给heights末尾加上0,这样可以在第一个for循环里弹出所有栈内的元素了,因为0小于栈内的所有元素。参考:
https://leetcode.com/problems/largest-rectangle-in-histogram/discuss/28959/My-concise-code-(20ms-stack-based-O(n))-one-trick-used

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值