【代码随想录二刷】Day60-单调栈-Go

代码随想录二刷Day60

今日任务

84.柱状图中最大的矩形
语言:Go

84. 柱状图中最大的矩形

链接:https://leetcode.cn/problems/largest-rectangle-in-histogram/

  1. 栈头到栈底的顺序:从大到小
  2. 原因:这样可以找到高柱子左右两边离的最近的矮柱子
  3. 开头加0的原因:如果数组是降序(8 6 4 2),没有开头的0,当i走到第二个元素6的时候,无法计算面积(没有left)
  4. 结尾加0的原因:如果数组是升序(2 4 6 8),没有结尾的0,无法走到计算面积的逻辑中,结果一直是0
  5. 计算面积的方法:每次以高柱子作为最右侧的矩形,从高向矮不断计算
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
}

版权声明:本文为weixin_38255385原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。