代码随想录二刷Day60
今日任务
84.柱状图中最大的矩形
语言:Go
84. 柱状图中最大的矩形
链接:https://leetcode.cn/problems/largest-rectangle-in-histogram/
- 栈头到栈底的顺序:从大到小
- 原因:这样可以找到高柱子左右两边离的最近的矮柱子
- 开头加0的原因:如果数组是降序(8 6 4 2),没有开头的0,当i走到第二个元素6的时候,无法计算面积(没有left)
- 结尾加0的原因:如果数组是升序(2 4 6 8),没有结尾的0,无法走到计算面积的逻辑中,结果一直是0
- 计算面积的方法:每次以高柱子作为最右侧的矩形,从高向矮不断计算
func largestRectangleArea(heights []int) int {
res := 0
stack := []int{0}
height := make([]int, len(heights) + 2)
height[0] = 0
for i := 0; i < len(heights); i++ {
height[i + 1] = heights[i]
}
height[len(height) - 1] = 0
for i := 1; i < len(height); i++ {
for height[i] < height[stack[len(stack) - 1]] { // <
mid := stack[len(stack) - 1]
stack = stack[:len(stack) - 1]
h := height[mid]
w := i - stack[len(stack) - 1] - 1 // -
if res < h * w {
res = h * w
}
}
stack = append(stack, i)
}
return res
}