单调栈方法解决最大矩形面积问题,数帽子问题,寻找第一个比自己大的数

摘自:详细介绍单调栈的理解和使用

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

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

以上是柱状图的示例,其中每个柱子的宽度为 1,给定的高度为 [2,1,5,6,2,3]。

图中阴影部分为所能勾勒出的最大矩形面积,其面积为 10 个单位。

示例:

输入: [2,1,5,6,2,3]
输出: 10

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/largest-rectangle-in-histogram
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        //动态规划
        if(heights.empty()){
            return 0;
        }
        //这是一个单调递增的栈,即从下到上都是递增的,栈顶元素最大,
        int res=0;
        heights.push_back(0);
        int len = heights.size();
        stack<int>mystack;
        for(int i=0;i<len;i++){
            while(!mystack.empty()&&heights[i]<heights[mystack.top()]){
                int num = mystack.top();
                mystack.pop();
                int h = heights[num];
                //int left = mystack.top();//这里不能这么写,原因是这里的栈可能是空,这样就会出现问题
                res = max(res,h*(mystack.empty()?i:(i-mystack.top()-1)));

            }
            mystack.push(i);
        }

        return res;
    }
};

数帽子问题也一样:
现有一条排好的队伍,从队首到队尾,队员们都戴着帽子,身高是无序的。假设每个人能看到队伍中在他前面的比他个子矮的人的帽子,(如果出现一个比这个人个子高的人挡住视线,那么此人不能看到高个子前面的任何人的帽子。)现在请计算出这个队伍中一共可以看到多少个帽子?例如给定数组为:[2,1,5,6,2,3](顺序为从队尾到队首)。

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        //动态规划
        if(heights.empty()){
            return 0;
        }
        //这是一个单调递增的栈,即从下到上都是递增的,栈顶元素最大,
        int res=0;
        heights.push_back(INT_MAX);
        int len = heights.size();
        stack<int>mystack;
        for(int i=0;i<len;i++){
            while(!mystack.empty()&&heights[i]>heights[mystack.top()]){
                int num = mystack.top();
                mystack.pop();
                int h = heights[num];
                //int left = mystack.top();//这里不能这么写,原因是这里的栈可能是空,这样就会出现问题
                // res = max(res,h*(mystack.empty()?i:(i-mystack.top()-1)));
                res += i-num-1;

            }
            mystack.push(i);
        }

        return res;
    }
};
给一个数组,返回一个大小相同的数组。返回的数组的第i个位置的值应当是,对于原数组中的第i个元素,至少往右走多少步,才能遇到一个比自己大的元素(如果之后没有比自己大的元素,或者已经是最后一个元素,则在返回数组的对应位置放上-1)。

例如给定数组为:[2,1,5,6,2,3]
返回数组应该为:[2,1,1,-1,1,-1]

其实这个问题本质上和数帽子问题是一样的,本质都是找到元素右边第一个比它大的数,代码稍作改动即可。

作者:edward_wang
链接:https://leetcode-cn.com/problems/largest-rectangle-in-histogram/solution/xiang-xi-jie-shao-dan-diao-zhan-de-li-jie-he-shi-y/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值