leetcode(84)柱状图中最大的矩形——栈

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

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

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

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

思路

来自leetcode@powcai
首先，要想找到第 i 位置最大面积是什么？

是以i 为中心，向左找第一个小于 heights[i] 的位置 left_i；向右找第一个小于于 heights[i] 的位置 right_i，即最大面积为 heights[i] * (right_i - left_i -1)，如下图所示：

所以，我们的问题就变成如何找 right_i 和 left_i？

当我们找 i 左边第一个小于 heights[i] 如果 heights[i-1] >= heights[i]其实就是和 heights[i-1] 左边第一个小于 heights[i-1] 一样。依次类推，右边同理。
那么根据官方栈解法，当我们维护一个单调递增的栈，直到遇到相邻柱子呈下降关系，也就是 a[i-1] > a[i]时，可以确定 i - 1 位置 的left_i为 i - 2（因为栈时单增的），right_i为i。得出面积为：

                        (i−（i - 1 )−1)×a[i-1].

对于栈中的每个元素，只要满足a[stack[j] > a[i] ，我们都对他们求面积，更新最大面积。(∵对于stack[j]来说 left_i 和 right_i 也是已知的）
那么我们不断地弹出栈顶元素，直到遇到 stack[j]stack[j] 满足a[stack[j]]≤a[i] ，得出面积为：

                   (i−stack[top−1]−1)×a[stack[top]].

当我们到达数组的尾部时，我们将栈中剩余元素全部弹出栈。在弹出每一个元素是，我们用下面的式子来求面积：

                  (stack[top]−stack[top−1])×a[stack[top]]，

其中，stack[top]表示刚刚被弹出的元素。因此，我们可以通过每次比较新计算的矩形面积来获得最大的矩形面积。

代码

class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
      stack<int> S;
      int maxArea = 0;
      int p = 0;
      while(p < heights.size()){
        if(S.empty()){
            S.push(p);
            ++p;
        }
        else {
            int t = S.top();
            if(heights[p] >= heights[t]){
                S.push(p);
                ++p;
            }
            else{
                int h = heights[S.top()];
                S.pop();
                int r = p;
                int l = S.empty() ? -1 : S.top();
                maxArea = max(maxArea,h * ( r - l - 1));
            }
        }
      }

      while(!S.empty()){
        int h = heights[S.top()];
                S.pop();
                int r = p;
                int l = S.empty() ? -1 : S.top();
                maxArea = max(maxArea,h * ( r - l - 1));
      }

      return maxArea;
    }
};