LeetCode0084.柱状图中最大的矩形 Go语言AC笔记

时间复杂度：O(n)，空间复杂度：O(n)

解题思路

应用单调栈存放索引，遍历元素比栈顶元素的高度小就计算矩形面积然后栈顶元素出栈，否则就是遍历的索引入栈。

计算矩形面积的方法为，宽度就是当前遍历的索引与栈顶元素的下一个元素之差减1，高度就是栈顶元素对应高度。

至于为什么这么算还是建议去看官方题解吧，这里就不多做赘述了。

AC代码

func largestRectangleArea(heights []int) (res int) {
    //获取下标为i的元素高度
    getHeight:=func(i int)int{
        //栈底和栈顶两个哨兵的高度为0
        if i==-1||i==len(heights){
            return 0
        }
        return heights[i]
    }
    stack:=make([]int,0,len(heights)+2)//单调栈，存放的是下标。容量大小要考虑两个哨兵
    stack=append(stack,-1)//栈底哨兵
    //必须遍历到栈顶哨兵，让栈中除栈底哨兵外所有元素弹出
    for i:=0;i<=len(heights);i++{
        //维持单调栈。当前遍历的元素小于栈顶元素了，需要弹出栈顶元素
        for getHeight(i)<getHeight(stack[len(stack)-1]){
            res=max(res,getHeight(stack[len(stack)-1])*(i-stack[len(stack)-2]-1))
            stack=stack[:len(stack)-1]
        }
        stack=append(stack,i)//入栈
    }
    return res
}
func max(i,j int)int{
    if i>j{
        return i
    }
    return j
}

感悟

第一次接触到单调栈，对单调栈的运作有了一定的认识，代码有几处细节也需要注意，总之做完这题收获颇丰！