Little-Prince

导航

84. 柱状图中最大的矩形(单调栈)

84. 柱状图中最大的矩形

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

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

 

 

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

 

 

 

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

示例:

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



思路:利用单调栈。
枚举「高」的方法:
首先我们枚举某一根柱子i作为高 h=heights[i];
随后我们需要进行向左右两边扩展,使得扩展到的柱子的高度均不小于 h。换句话说,我们需要找到左右两侧最近的高度小于 h 的柱子,这样这两根柱子之间(不包括其本身)的所有柱子高度均不小于 h,并且就是 i 能够扩展到的最远范围。
 
那么我们先来看看如何求出一根柱子的左侧且最近的小于其高度的柱子。
对于两根柱子 j0​ 以及 j1​,如果 j0<j1​ 并且 heights[j0]≥heights[j1],那么对于任意的在它们之后出现的柱子 i(j1<i),j0 一定不会是 i 左侧且最近的小于其高度的柱子。
换句话说,如果有两根柱子 j0和 j1,其中 j0 在 j1​ 的左侧,并且 j0 的高度大于等于 j1,那么在后面的柱子 i向左找小于其高度的柱子时,j1 会挡住j0​,j0 就不会作为答案了。
这样一来,我们可以对数组从左向右进行遍历,同时维护一个「可能作为答案」的数据结构,其中按照从小到大的顺序存放了一些 j 值。根据上面的结论,如果我们存放了 j0,j1,⋯ ,js​,那么一定有 height[j0]<height[j1]<⋯<height[js],因为如果有两个相邻的 j 值对应的高度不满足 < 关系,那么后者会「挡住」前者,前者就不可能作为答案了。
当我们枚举到第 i 根柱子时,我们的数据结构中存放了 j0,j1,⋯ ,js,如果第 i 根柱子左侧且最近的小于其高度的柱子为 ji​,那么必然有
height[j0]<height[j1]<⋯<height[ji]<height[i]≤height[ji+1]<⋯<height[js]
当我们枚举到 i+1 时,原来的 i 也变成了 j 值,因此 i 会被放入数据结构。由于所有在数据结构中的 j 值均小于 i,那么所有高度大于等于 height[i] 的 j 都不会作为答案,需要从数据结构中移除。而我们发现,这些被移除的 j 值恰好就是
ji+1,⋯ ,js
这样我们在枚举到第 i 根柱子的时候,就可以先把所有高度大于等于 height[i]的 j 值全部移除,剩下的 j 值中高度最高的即为答案。在这之后,我们将 i 放入数据结构中,开始接下来的枚举。此时,我们需要使用的数据结构也就呼之欲出了,它就是栈。

栈中存放了 j 值。从栈底到栈顶,j 的值严格单调递增,同时对应的高度值也严格单调递增;

当我们枚举到第 i 根柱子时,我们从栈顶不断地移除 height[j]≥height[i] 的 j 值。在移除完毕后,栈顶的 j 值就一定满足 height[j]<height[i],此时 j 就是 i 左侧且最近的小于其高度的柱子。
这里会有一种特殊情况。如果我们移除了栈中所有的 j 值,那就说明 i 左侧所有柱子的高度都大于 height[i],那么我们可以认为 i左侧且最近的小于其高度的柱子在位置 j=−1,它是一根「虚拟」的、高度无限低的柱子。这样的定义不会对我们的答案产生任何的影响,我们也称这根「虚拟」的柱子为「哨兵」。
 
我们再将 i 放入栈顶。

栈中存放的元素具有单调性,这就是经典的数据结构「单调栈」了。
 
我们在对位置 i 进行入栈操作时,确定了它的左边界。从直觉上来说,与之对应的我们在对位置 i 进行出栈操作时可以确定它的右边界!当位置 i 被弹出栈时,说明此时遍历到的位置 i0​ 的高度小于等于 height[i],并且在 i0​ 与 i 之间没有其他高度小于等于 height[i] 的柱子。这是因为,如果在 i 和 i0​ 之间还有其它位置的高度小于等于 height[i] 的,那么在遍历到那个位置的时候,i 应该已经被弹出栈了。所以位置 i0​ 就是位置 i 的右边界。
等等,我们需要的是「一根柱子的左侧且最近的小于其高度的柱子」,但这里我们求的是小于等于,那么会造成什么影响呢?答案是:我们确实无法求出正确的右边界,但对最终的答案没有任何影响。这是因为在答案对应的矩形中,如果有若干个柱子的高度都等于矩形的高度,那么最右侧的那根柱子是可以求出正确的右边界的,而我们没有对求出左边界的算法进行任何改动,因此最终的答案还是可以从最右侧的与矩形高度相同的柱子求得的。读者可以仔细思考一下这一步。
在遍历结束后,栈中仍然有一些位置,这些位置对应的右边界就是位置为 n 的「哨兵」。我们可以将它们依次出栈并更新右边界,也可以在初始化右边界数组时就将所有的元素的值置为 n。
 
时间复杂度 o(N), 空间复杂度 o(N)。
代码:
class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        int N = heights.size();
        vector<int> left(N);//定义了 N 个 int 元素的向量
        vector<int> right(N,N);//定义了 N 个 int 元素的向量,初值为 N
        stack<int>mono_stack;
        int i, pos;
        for(i = 0; i < N; i++)
        {
            if(mono_stack.empty())
            {
                mono_stack.push(i);
                left[i] = -1;
            }
            else
            {
                while(!mono_stack.empty())
                {
                   pos = mono_stack.top();
                   if(heights[pos]>=heights[i])
                   {
                        mono_stack.pop();
                        right[pos] = i;
                   }
                   else
                       break; 
                }
                left[i] = mono_stack.empty() ? -1 : mono_stack.top();
                mono_stack.push(i);              
            }
        }
        int ans = 0;
        for(i = 0; i < N; i++)
        {
            ans = max(ans, heights[i]*(right[i]-left[i]-1));
        }
        return ans;
    }
};

 

 
 

posted on 2020-08-03 23:52  Little-Prince  阅读(189)  评论(0编辑  收藏  举报