leetcode 84.柱状图中最大的矩形 Java

题目链接

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

描述

给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。

求在该柱状图中，能够勾勒出来的矩形的最大面积。

示例

输入: [2,1,5,6,2,3]
输出: 10

代码

比较推荐的解法是单调栈加哨兵，由于题目链接里有很详细且优秀的解释，所以如果对代码有疑问，可以直接点击题目链接去看一下。
不过代码也是比较容易理解的，之所以添加“哨兵”是为了减少空栈的判定情况。

class Solution {
    public int largestRectangleArea(int[] heights) {
        if (heights == null || heights.length == 0) {
            return 0;
        }

        //为了避免繁琐的判断，添加哨兵
        int[] arr = new int[heights.length + 2];
        System.arraycopy(heights, 0, arr, 1, heights.length);

        Deque<Integer> stack = new ArrayDeque<>();
        stack.push(0);
        int res = 0;
        int height = 0;
        int width = 0;
        for (int i = 1; i < arr.length; i++) {
            while (arr[stack.peek()] > arr[i]) {
                height = arr[stack.pop()];
                width = i - stack.peek() - 1;
                res = Math.max(res, height * width);
            }

            stack.push(i);
        }

        return res;
    }
}