注:
题目:
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
示例 1:
输入:heights = [2,1,5,6,2,3]
输出:10
解释:最大的矩形为图中红色区域,面积为 10
示例 2:
输入: heights = [2,4]
输出: 4
提示:
1 <= heights.length <=105
0 <= heights[i] <= 104
题解:
思路与算法
解法:依次遍历柱形的高度,对于每一个高度分别向两边扩散,求出以当前高度为矩形的最大宽度多少。
为此,我们需要:
- 左边看一下,看最多能向左延伸多长,找到大于等于当前柱形高度的最左边元素的下标;
- 右边看一下,看最多能向右延伸多长;找到大于等于当前柱形高度的最右边元素的下标
因此,我们采用单调栈来解决该题。
操作规则(下面都以单调递增栈为例)
- 如果新的元素比栈顶元素大,就入栈
- 如果新的元素较小,那就一直把栈内元素弹出来,直到栈顶比新元素小
效果:
- 栈内的元素是递增的
- 当元素出栈时,说明这个新元素是出栈元素向后找第一个比其小的元素
- 当元素出栈后,说明新栈顶元素是出栈元素向前找第一个比其小的元素
举个例子,配合下图,现在索引在 6 ,栈里是 1 5 6 。接下来新元素是 2 ,那么 6 需要出栈。
当 6 出栈时,右边 2 代表是 6 右边第一个比 6 小的元素。
当 6 出栈时,5 成为新的栈顶,那么 5 就是 6 左边第一个比 6 小的元素。
因此找到了以6为高度的最大矩形的面积,
以上思路 需要考虑两种特殊的情况:
- 弹出一个元素,栈为空,此时无法取得栈顶元素;
- 遍历完成以后,栈中还有元素;
为此可以我们可以在输入数组的两端加上两个高度为 0 (或者是 0.5,只要比 1 严格小都行)的柱形,可以回避上面这两种分类讨论。
这两个站在两边的柱形有一个很形象的名词,叫做哨兵(Sentinel)。
有了这两个柱形:
- 左边的柱形(第 1 个柱形)由于它一定比输入数组里任何一个元素小,它肯定不会出栈,因此栈一定不会为空;
- 右边的柱形(第 2 个柱形)也正是因为它一定比输入数组里任何一个元素小,它会让所有输入数组里的元素出栈(第 1 个哨兵元素除外)。
class Solution {
public:
int largestRectangleArea(vector<int>& heights) {
stack<int> restack;
//在heights前后补零
heights.insert(heights.begin(),0);
heights.push_back(0);
int maxspace=0;
for(int i=0;i<heights.size();i++){
while(!restack.empty()&&heights[restack.top()]>heights[i]){
//记录当前栈顶元素cur的高度
int cur=restack.top();
//弹出栈顶元素
restack.pop();
//新栈顶为cur前面第一个比cur小的元素
//由于在数组开头补了0,因此restack永不为空,restak.top()不会出错
int left=restack.top()+1;
//i为cur后面第一个比cur小的元素
int right=i;
//计算以cur为高的矩形的最大面积
maxspace=max(maxspace,(right-left)*heights[cur]);
}
restack.push(i);
}
return maxspace;
}
};