【刷题】[LeetCode#84] 单调栈法求解柱状图中最大矩形面积

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

题目

给定 n 个非负整数，用来表示柱状图中各个柱子的高度。每个柱子彼此相邻，且宽度为 1 。
求在该柱状图中，能够勾勒出来的矩形的最大面积。

以上是柱状图的示例，其中每个柱子的宽度为 1，给定的高度为 [2,1,5,6,2,3]。

图中阴影部分为所能勾勒出的最大矩形面积，其面积为 10 个单位。

解法

本题目可以用分治法或利用单调栈来求解。要想得到最大的矩形的面积，就要求得可以完全覆盖住柱状图中每个柱子的最大矩形的面积。以上图为例，能覆盖住各个柱子的最大矩形面积分别为2=2*1，6=1*6，10=5*2，6=6*1，8=2*4，3=3*1。从中取最大值10即为所求。
以上图中的5为例，最大矩形的高度为5，宽度为其两侧第一个比它低的矩形的距离，也就是上图中1和右边的2之间的距离。为了求得这个宽度，可以利用单调栈。
定义单调栈的行为（以单增栈为例）：

• 入栈：当栈为空，直接入栈；否则若栈顶元素小于要入栈的元素，将栈顶弹出，重复判断直到栈空或栈顶元素小于要入栈的元素。
• 出栈、取栈顶、判栈空等都与普通栈相同。

这样，在元素p入栈时，如果元素q被弹出，则p是q之后第一个比它小的元素，弹出后的栈顶元素是q之前第一个比它小的元素。这样在每个元素弹出时即求得对应矩形的宽度，本题得解，时间复杂度为O(n)。

特殊情况讨论

因为有可能出现连续的相同元素，所以需要讨论单调栈是否应该是严格单调的。假设满足严格单调，则以{3, 3, 3, 2, 4}这个柱状图为例，当第二个和第三个3入栈时，都需要将栈顶的3弹出，此时每个3对应的宽度均为2，而实际上最大宽度应当为3，这是不合理的。如果不必满足严格单调，那么在3入栈时不必弹栈，而在2入栈时再统一弹栈，这样虽然先弹出的3的宽度会偏小，但最后一个弹出的3的宽度一定是准确的，所以这样不会带来问题。
结论：单调栈不必满足元素的严格单调（也就是说相邻元素可以相等）。

代码

Java：（为了方便求得柱的距离，此处入栈的不是柱的高度，而是柱的下标，柱的高度可以通过下标访问数组得到）

class Solution {
    public int largestRectangleArea(int[] heights) {
        Stack<Integer> stack = new Stack<>();
        int max = 0;
        for (int i = 0; i < heights.length; i++) {
            while (!stack.empty() && heights[stack.peek()] > heights[i]) {
                int top = stack.pop();
                int width = stack.empty() ? i : (i - stack.peek() - 1);
                max = Math.max(max, heights[top] * width);
            }
            stack.push(i);
        }
        while (!stack.empty()) {
            int top = stack.pop();
            int width = stack.empty() ? heights.length : (heights.length - stack.peek() - 1);
            max = Math.max(max, heights[top] * width);
        }
        return max;
    }
}