力扣84.柱状图中最大的矩形 c++

一、题目描述

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

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

二、示例

三、分析

思路（从左向右看）：

   （1） 如果下一个比当前高，说明当前柱子可以向后扩展，即当前高度可以和后面的柱子组成更大的矩形；

   （2） 如果下一个比当前小，说明当前柱子的高度不能再向后扩展，而由上一条可知，当前柱子的上一条也比自己矮，因此可以确定以当前柱子为高的最大面积；

                当前柱子能够确定，如果前一个柱子的高度也比下一个高，那么前面的柱子也就能够确定了；所以，后面的柱子先确定面积，可以使用栈来解决。

    （3）栈中存柱子的下标，可以通过下标再heights数组中获取高度。

步骤：遍历heights数组，如果栈为空，直接入栈；

           如果栈不为空，比较栈顶top和当前柱子（下标i）的高度，

                      如果比栈顶高，入栈；

                      如果没有栈顶高，栈顶出栈，栈顶的面积为（(i-top)*heights[top]）

四、代码

class Solution {
public:

// //暴力：超时
//     int largestRectangleArea(vector<int>& heights) {
//         int max=0;
//         int len;
//         for(int i=heights.size()-1;i>=0;i--){
//             len=heights[i];
//             for(int j=i;j>=0;j--){
//                 if(heights[j]<len)len=heights[j];
//                 if(i==j)h[i][j]=heights[i];
//                 else h[i][j]=(i-j+1)*len;
//                 if(h[i][j]>max)max=h[i][j];
//             }
//         }
//         return max;
//     }
    int largestRectangleArea(vector<int>& heights) {
        int ans=0;
        stack<int>sta;
        int i;
        for(i=0;i<heights.size();i++){
            if(!sta.size())sta.push(i);
            else if(heights[sta.top()]<=heights[i])sta.push(i);
            else{
                while(!sta.empty()&&heights[sta.top()]>heights[i]){
                    int top=sta.top();
                    sta.pop();
                    int haha=0;
                    if(!sta.empty())haha=heights[top]*(top-sta.top()-1);
                    else haha=heights[top]*top;
                    ans=ans>(i-top)*heights[top]+haha?ans:(i-top)*heights[top]+haha;
                }
                sta.push(i);
            }      
        }
        while(!sta.empty()){
            // ans=ans>(i-sta.top())*heights[sta.top()]?ans:(i-sta.top())*heights[sta.top()];
            // sta.pop();
            int top=sta.top();
            sta.pop();
            int haha=0;
            if(!sta.empty())haha=heights[top]*(top-sta.top()-1);
            else haha=heights[top]*top;
            ans=ans>(i-top)*heights[top]+haha?ans:(i-top)*heights[top]+haha;
        }
        return ans;
    }
};