题目
给定 n 个非负整数,用来表示柱状图中各个柱子的高度。每个柱子彼此相邻,且宽度为 1 。
求在该柱状图中,能够勾勒出来的矩形的最大面积。
以上是柱状图的示例,其中每个柱子的宽度为 1,给定的高度为 [2,1,5,6,2,3]。
图中阴影部分为所能勾勒出的最大矩形面积,其面积为 10 个单位。
示例:
输入: [2,1,5,6,2,3] 输出: 10
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/largest-rectangle-in-histogram
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
- 对于一个高度,如果能得到向左和向右的边界
- 那么就能对每个高度求一次面积
- 遍历所有高度,即可得出最大面积
首先是暴力枚举+剪枝,每一根柱子都向左右找到第一根矮于自己的柱子(被卡脖子),然后结合自身的高度计算。这种做法的剪枝策略就是当边缘柱子和自身高度相同的时候,不需要再次计算边缘柱子的最左最右了。我看见有人的操作是直接让边缘柱子的height–。真是非常机智的操作,但是它的正确性还是有点难以理解的。
其次的方法就是所谓题解的“单调栈”。看起来像是一个全新的数据结构,其实不然。只不过是上面暴力方法用空间换取时间的一种策略。从左往右当高度增加的时候,就不断地压栈、计算面积,如果高度减少了,就开始弹栈直到栈顶元素小于当前元素。每次在弹栈的时候计算能覆盖的最大面积。
用这张复制来的图片可能能帮助理解。
这样做可以保证两点:栈内的元素是递增的;当元素出栈时,说明这个新元素是出栈元素向后找第一个比其小的元素。
如此一来就可以根据元素所在的位置计算面积。
代码1——暴力+剪枝
int largestRectangleArea(int* heights, int heightsSize) {
int max = 0, sum = 0;
for (int i = 0; i < heightsSize; i++) {
if (heights[i] != 0) { // 非0才计算
sum = heights[i];
for(int j = i - 1; j >= 0; j--){
if(heights[j] >= heights[i]) {
sum += heights[i];
} else {
break;
}
if(heights[i] == heights[j]) {
heights[j]--; // 不影响结果的情况下的剪枝
}
}
for(int k = i + 1; k < heightsSize; k++){
if(heights[k] >= heights[i]) {
sum += heights[i];
} else {
break;
}
if(heights[k] == heights[i]) {
heights[k]--; // 不影响结果的情况下的剪枝
}
}
if(max < sum) {
max = sum;
}
}
}
return max;
}
运行结果
代码2——单调栈
int largestRectangleArea(int* heights, int heightsSize){
int top = -1;
int area, i;
int maxarea = 0;
int *stack = (int *)malloc(sizeof(int) * (heightsSize + 2));
int *buff = (int *)malloc(sizeof(int) * (heightsSize + 2));
// 在前面加哨兵
buff[0] = 0;
for (int i = 1; i <= heightsSize; i++) {
buff[i] = heights[i - 1];
}
// 在最后加哨兵
buff[heightsSize + 1] = 0;
stack[++top] = 0;
for (i = 1; i < heightsSize + 2; i++) {
while (top > 0 && buff[i] < buff[stack[top]]) {
area = (i - stack[top - 1] - 1) * buff[stack[top]];
maxarea = maxarea > area ? maxarea : area;
top--;
}
stack[++top] = i;
}
return maxarea;
}
运行结果
感谢大家的支持和点赞。