1 题目描述
给出n个数字,代表直方图的条高,直方图每一条的宽度为1,请计算直方图中最大矩形的面积
例如:
给出的高度 =[2,1,5,6,2,3],
返回10.
2 解题思路
目的:用height[]构造一个升序栈,构造过程中计算面积;
如果当前height[i]大于栈顶元素,则入栈;
若小于栈顶元素,则将栈顶元素弹出并做记录弹出几次,并计算以弹出元素作为高度的面积,留下
最大值ret,直到满足height[i]大于栈顶元素,再将弹出的元素以height[i]重新入栈;
过程为 :
- 2入栈;目前栈为{2}
- 1与2比较,不满足升序,则2弹出,记录count=1;ret=2*1;1代替2再次入栈,然后当前1入栈;目前栈为{1,1}
- 5入栈,满足升序,6入栈满足升序;目前栈为{1,1,5,6,}
- 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}
- height[5]=3入栈;形成升序栈{1,1,2,2,2,3}
- 最后按照升序栈继续维护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