给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
以上是柱状图的示例,其中每个柱子的宽度为 1,给定的高度为 [2,1,5,6,2,3]。
图中阴影部分为所能勾勒出的最大矩形面积,其面积为 10 个单位。
示例:
输入: [2,1,5,6,2,3]
输出: 10
这道题用单调栈,先说说单调栈有啥用吧。
单调栈,存在栈中的数据是递增或者递减数据的下标,这样我就很容易知道以当前高度为基准,左右两边最近的比他要大的(例如:一排人,求每个人看到的脑袋顶的数量的总和,其实就是找左右两边第一个大于他们的),左右两边最近的比他要小的(例如:上面这个题目,左右两边比他小的就不能构成矩形)。
相似的题目如接雨水,不能用这样的思想,因为他不是以当前高度为基准,他是要寻找当前高度左右两边最高的。
代码如下:
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
if (heights.size() == 0)
return 0;
if (heights.size() == 1)
return heights[0];
//以下通过加哨兵,方便判断
vector<int> new_heights(heights.size() + 2, 0);
copy(heights.begin(), heights.end(), new_heights.begin() + 1);
int ans = 0;
stack<int> sta;
sta.push(0);
for (int i = 1; i < new_heights.size(); i++) {
if (new_heights[i] > new_heights[sta.top()])
sta.push(i);
else if(new_heights[i] == new_heights[sta.top()]) {
sta.pop();
sta.push(i);
}
else {
while (new_heights[i] < new_heights[sta.top()]) {
int height_temp = new_heights[sta.top()];
sta.pop();
ans = max(ans, height_temp * (i - sta.top() - 1));
}
sta.push(i);
}
}
return ans;
}
};
上面代码还有个比较重要的点,哨兵,他就是站在单调栈的左右两边,死死地夹紧这个单调栈,让他不会为空,也让他最后全部元素都安全出去,就像是哨兵一样,看着中间的元素,尽职尽责。