算法训练 Day 60

LeetCode 84. 柱状图中最大的矩形

题目链接：84. 柱状图中最大的矩形

思路：本题与著名的经典题目接雨水思路一致，但实际上还要更复杂一些，接雨水是找每个柱子左右两边第一个大于该柱子高度的柱子，而本题是找每个柱子左右两边第一个小于该柱子的柱子。

关于这道题，我们应该明确一点，遍历每个列的高度，是要以当前高度为基准，寻找最大的宽度，组成最大的矩形面积。做法是要找左边第一个小于当前高度的下标left，再找右边第一个小于当前高度的下标right，那宽度就是这两个下标之间的距离了。但是要排除这两个下标，所以是right - left - 1。用单调栈就可以很方便的确定这两个边界。在单调栈构造的过程中，依旧是三种情况：

• 当前遍历的元素heights[i]大于栈顶元素heights[st.top()]的情况
  直接将当前遍历元素入栈。
• 当前遍历的元素heights[i]等于栈顶元素heights[st.top()]的情况
  因为单调栈想找出的是小于当前高度的下标（知道最大矩形的边界在哪），所以等于的时候同第一种情况相同，入栈。
• 当前遍历的元素heights[i]小于栈顶元素heights[st.top()]的情况
  将栈顶元素弹出，计算以当前最大高度的矩形形成的面积，与遍历过程中记录的res进行比较，取其中的较大者。并将当前遍历元素持续与新的栈顶元素比较（之后有可能有高度更高的矩形能拼接上），直到遇到上述两种情况或者该元素成为栈顶元素为止。

go版本：

func largestRectangleArea(heights []int) int {
    n := len(heights)
    res := 0
    nums := []int{0}
    // 在输入数组的两端加上两个高度为0的“哨兵”
    // “哨兵”可以避免讨论弹出栈顶元素后栈为空的情况, 以及遍历结束之后栈中还存在变量
    heights = append(nums, heights...)
    heights = append(heights, 0)
    stack := make([]int, 1)
    stack[0] = 0
    cur_height := 0
    cur_width := 0

    for i:=0; i<n+2; i++ {
    	// 当前遍历的元素T[i]小于栈顶元素T[st.top()]
        for heights[stack[len(stack)-1]] > heights[i] {
            temp := stack[len(stack)-1]
            stack = stack[:len(stack)-1]
            cur_height = heights[temp]
            cur_width = i - stack[len(stack)-1] - 1
            res = max(res, cur_height * cur_width)
        }
        stack = append(stack, i)
    }
    return res
}

func max(i, j int) int {
    if i>j {
        return i
    }
    return j
}