leetcode, LC65:直方图中的最大矩形

1 题目描述

给出n个数字，代表直方图的条高，直方图每一条的宽度为1，请计算直方图中最大矩形的面积

例如：

给出的高度 =[2,1,5,6,2,3],

返回10.

2 解题思路

目的：用height[]构造一个升序栈,构造过程中计算面积；

如果当前height[i]大于栈顶元素，则入栈；

若小于栈顶元素，则将栈顶元素弹出并做记录弹出几次，并计算以弹出元素作为高度的面积，留下

最大值ret，直到满足height[i]大于栈顶元素，再将弹出的元素以height[i]重新入栈；

过程为 ：

1. 2入栈；目前栈为{2}
2. 1与2比较，不满足升序，则2弹出，记录count=1；ret=2*1；1代替2再次入栈，然后当前1入栈；目前栈为{1,1}
3. 5入栈，满足升序，6入栈满足升序；目前栈为{1,1,5,6,}
4. height[4]=2，即将入栈，由于2小于栈顶元素6，则6弹出，count=1，ret=max（2,6）=6；2小于5,5弹出，count=2,ret=max(6,2*5)=10;6和5 重新以2入栈，然后height[4]=2入栈；目前栈为{1,1,2,2,2}
5. height[5]=3入栈；形成升序栈{1,1,2,2,2,3}
6. 最后按照升序栈继续维护ret直至栈为空，max(ret,31,22,23,24*,15,16)=10;

3 代码实现

class Solution {
public:
    /**
     * 
     * @param height int整型vector 
     * @return int整型
     */
    int largestRectangleArea(vector<int>& height) {
        // write code here
        int area = 0;
        stack<int> stk;
        height.push_back(0);
        for(int i = 0; i < height.size(); i++){
            if(stk.empty() || height[i] >= stk.top()){
                stk.push(height[i]);
            }
            else{
                int count = 0;
                while(!stk.empty() && height[i] < stk.top()){
                    count++;
                    area = max(area, count * stk.top());
                    stk.pop();
                }
                while(count--){
                    stk.push(height[i]);
                }
                stk.push(height[i]);
            }
        }
        return area;
    }
};

4 运行结果

运行时间：7ms
占用内存：1124k