单调栈题目：柱状图中最大的矩形

题目

标题和出处

标题：柱状图中最大的矩形

出处：84. 柱状图中最大的矩形

难度

7 级

题目描述

要求

给定整数数组 $\text{heights}$ 表示柱状图中各个柱子的高度，其中每个柱子的宽度为 $\text{1}$，返回在该柱状图中的矩形的最大面积。

示例

示例 1：

输入：$\text{heights = [2,1,5,6,2,3]}$
输出：$\text{10}$
解释：上图为一个柱状图，其中每个柱子的宽度为 $\text{1}$。最大的矩形为图中红色区域，面积为 $\text{10}$。

示例 2：

输入：$\text{heights = [2,4]}$
输出：$\text{4}$

数据范围

• $\text{1}\le \text{heights.length}\le {\text{10}}^{\text{5}}$
• $\text{0}\le \text{heights[i]}\le {\text{10}}^{\text{4}}$

解法

思路和算法

为了得到柱状图中最大的矩形，对于下标 $i$，需要找到最大的下标 $j$ 和最小的下标 $k$，满足 $j<i<k$ 且 $\text{heights}[j]<\text{heights}[i]$ 和 $\text{heights}[k]<\text{heights}[i]$，则存在一个宽度为 $k-j-1$、高度为 $\text{heights}[i]$ 的矩形，计算该矩形的面积并更新矩形的最大面积。

直观的做法是遍历柱状图中的每个柱子，对于每个下标 $i$，向两边遍历寻找对应的下标 $j$ 和 $k$，得到以 $\text{heights}[i]$ 为高的最大矩形宽度并计算矩形的面积。假设数组 $\text{heights}$ 的长度是 $n$，该做法的时间复杂度是 $O(n^2)$，会超出时间限制，因此需要优化。

创建两个长度为 $n$ 的数组 $\text{left}$ 和 $\text{right}$，对于每个下标 $i$，$\text{left}[i]$ 和 $\text{right}[i]$ 分别记录对应的下标 $j$ 和 $k$。初始时，$\text{left}$ 的全部元素为 $-1$，$\text{right}$ 的全部元素为 $n$。这道题中可以假设 $\text{heights}[-1]=\text{heights}[n]=-\infty$。在遍历数组 $\text{heights}$ 的过程中填入 $\text{left}$ 和 $\text{right}$ 的值。

为了得到 $\text{left}$ 和 $\text{right}$ 的值，可以使用单调栈，单调栈存储数组 $\text{heights}$ 的下标，满足从栈底到栈顶的下标对应的元素单调递增。

从左到右遍历数组 $\text{heights}$，当遍历到下标 $i$ 时，进行如下操作：

1. 如果栈不为空且栈顶下标对应的元素大于等于 $\text{heights}[i]$，则将栈顶下标出栈，并将栈顶下标对应的 $\text{right}$ 值设为 $i$，重复该操作直到栈为空或者栈顶下标对应的元素小于 $\text{heights}[i]$；

2. 如果栈不为空，则栈顶下标对应的元素小于 $\text{heights}[i]$，因此将 $i$ 对应的 $\text{left}$ 值设为栈顶下标；

3. 将 $i$ 入栈。

上述操作中，下标 $i$ 入栈的条件是栈为空或者栈顶下标对应的元素小于 $\text{heights}[i]$，因此可以保证从栈底到栈顶的下标对应的元素单调递增。

遍历结束之后，对于每个下标 $i$，令 $j=\text{left}[i]$，$k=\text{right}[i]$，则有 $\text{heights}[j]<\text{heights}[i]$，$\text{heights}[k]\le \text{heights}[i]$。如果 $\text{heights}[k]<\text{heights}[i]$，则高度为 $\text{heights}[i]$ 的最大矩形宽度即为 $k-j-1$。如果 $\text{heights}[k]=\text{heights}[i]$，则高度为 $\text{heights}[i]$ 的最大矩形宽度大于 $k-j-1$，但是在遍历到下标 $k$ 时可以得到高度为 $\text{heights}[k]$ 的矩形宽度为 $\text{right}[k]-\text{left}[k]-1$，该宽度大于 $k-j-1$。由于一定存在一个柱子，该柱子右边没有和该柱子高度相同的柱子，因此在遍历到该柱子时即可得到以该柱子的高度为矩形高度的矩形最大宽度，并计算得到矩形的最大面积。

对于下标 $i$，高度为 $\text{heights}[i]$ 的最大矩形的宽度为 $\text{right}[i]-\text{left}[i]-1$（根据上述分析，可能存在 $k>i$ 满足 $\text{heights}[k]=\text{heights}[i]$，但是这里仍然认为该宽度是最大矩形的宽度），面积为 $(\text{right}[i]-\text{left}[i]-1)\times \text{heights}[i]$。遍历 $0\le i<n$，即可得到柱状图中的矩形的最大面积。

考虑一个例子：$\text{heights}=[2,1,4,4]$。对应的 $\text{left}=[-1,-1,1,1]$，$\text{right}=[1,4,3,4]$，矩形的最大面积是 $8$。

对于下标 $0$，最大矩形的宽度为 $1-(-1)-1=1$，高度为 $2$，面积为 $2$。

对于下标 $1$，最大矩形的宽度为 $4-(-1)-1=4$，高度为 $1$，面积为 $4$。

对于下标 $2$，最大矩形的宽度为 $3-1-1=1$，高度为 $4$，面积为 $4$。注意 $\text{heights}[2]=\text{heights}[3]$，因此最大矩形的宽度应该为 $2$，但是在下标 $2$ 处计算的最大矩形的宽度为 $1$，在下标 $3$ 处将会计算到最大矩形的宽度为 $2$。

对于下标 $3$，最大矩形的宽度为 $4-1-1=2$，高度为 $4$，面积为 $8$。

代码

class Solution {
    public int largestRectangleArea(int[] heights) {
        int length = heights.length;
        int[] left = new int[length];
        int[] right = new int[length];
        Arrays.fill(left, -1);
        Arrays.fill(right, length);
        Deque<Integer> stack = new ArrayDeque<Integer>();
        for (int i = 0; i < length; i++) {
            int height = heights[i];
            while (!stack.isEmpty() && heights[stack.peek()] >= height) {
                right[stack.pop()] = i;
            }
            if (!stack.isEmpty()) {
                left[i] = stack.peek();
            }
            stack.push(i);
        }
        int area = 0;
        for (int i = 0; i < length; i++) {
            area = Math.max(area, (right[i] - left[i] - 1) * heights[i]);
        }
        return area;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $\text{heights}$ 的长度。需要遍历数组 $\text{heights}$ 一次，每个下标最多入栈和出栈各一次。

• 空间复杂度：$O(n)$，其中 $n$ 是数组 $\text{heights}$ 的长度。空间复杂度主要取决于两个数组 $\text{left}$ 和 $\text{right}$ 以及栈空间，两个数组的长度都是 $n$，栈内元素个数不会超过 $n$。