给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
以上是柱状图的示例,其中每个柱子的宽度为 1,给定的高度为 [2,1,5,6,2,3]。
图中阴影部分为所能勾勒出的最大矩形面积,其面积为 10 个单位。
示例:
输入: [2,1,5,6,2,3]
输出: 10
思路:
简单的想法就是暴力法,对每个柱子左右扩展,知道碰到即目前较矮的,计算相应矩形面积,用一个变量记录最大面积,时间复杂度为O(n^2)
用栈做:维护一个递增的栈,每次比较栈顶与当前元素。如果当前元素大于栈顶元素,将i++入栈,否则合并现有栈,取最大面积,直到栈顶元素小于当前元素。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
stack<int> s;
heights.push_back(0);
int res = 0;
for(int i = 0; i < heights.size(); ) {
if(s.empty() || heights[i] > heights[s.top()]) {
s.push(i++);
} else {
int temp = s.top();
s.pop();
res = max(res, heights[temp] * (s.empty() ? i: i - s.top() - 1));
}
}
return res;
}
};