LeetCode(82) Largest Rectangle in Histogram

题目如下:

Given n non-negative integers representing the histogram's bar height where the width of each bar is 1, find the area of largest rectangle in the histogram.

Above is a histogram where width of each bar is 1, given height = [2,1,5,6,2,3].

The largest rectangle is shown in the shaded area, which has area = 10 unit.

For example,
Given height = [2,1,5,6,2,3],
return 10.

分析如下：

O(N)的作法比较不好想，google之得到思路如下，使用stack。思路也不太容易自己想出来。

首先，作为铺垫，想一个比较好想的思路，这里摘抄一段不错的解析，来自felixfang的blog: 

最大矩形的高度毫无疑问必然和某一个立柱的高度相等，或者说，最大矩形必然包含了某一个立柱的全部。

遍历所有立柱，对当前立柱 i，以其高度左右扩展，看看以当前立柱 i 的高度最多能包含进多大的矩形面积。最后选出最大的总面积即可。

所以，每个当前立柱i都尽可能的向左或者向右扩展，计算以当前立柱的高度为最终高度的矩形的面积。。在向左扩展的时候，如果遇到立柱比当前立柱i的高度更高，则继续

扩展，如果更低，则停止(此时更低的这个立柱不计入当前矩形面积的计算)，向右扩展同理。

然后，看看stack的作法，总得来说，stack帮助你维护一个上面这段描述的扩展图，使得每个扩展图对应的 S 能够被计算出来。

细节的解释还是摘抄一段felixiafang的解释：

height的内容是 [5,6,7,8,3]，特点是除了最后一个，前面全部保持递增，且最后一个立柱的高度小于前面所有立柱高度。
对于这种特点的柱状图，如果使用上面所说的“挨个使用每一个柱状图的高度作为矩形的高度，求面积”的方法，还需要用嵌套循环吗？
我们知道除了最后一个，从第一个到倒数第二个立柱的高度都在升高，那么如果挨个使用每一个柱的高度作为矩形的高度，那么依次能得到的矩形的宽度就可以直接算出来： 使用5作为高度可以使用前四个立柱组成 4*5的矩形，高度6可以组成3*6的矩形... 因此只需要遍历一次，选出最大面积即可。
对于这种类型的柱状图，最大矩形面积的时间复杂度是O(n)。
我们将这种特点的柱状图称为“波峰图”。
下面介绍新的解法的步骤：
(1) 在height尾部添加一个0，也就是一个高度为0的立柱。作用是在最后也能凑成上面提的那种“波峰图”。
(2) 定义了一个stack，然后遍历时如果height[i] 大于stack.top()，进栈。反之，出栈直到栈顶元素小于height[i]。
由于出栈的这些元素高度都是递增的，我们可以求出这些立柱中所围成的最大矩形。更妙的是，由于这些被弹出的立柱处于“波峰”之上(比如弹出i 到 i+k，那么所有这些立柱的 高度都高于 i-1和 i+k+1的高度)，因此，如果我们使用之前所提的“左右延伸找立柱”的思路解，以这些立柱的高度作为整个矩形的高度时，左右延伸出的矩形所包含的立柱不会 超出这段“波峰”，因为波峰外的立柱高度都比他们低。“波峰图”其实就是求解最大矩形的“孤岛”，它不会干扰到外部。
(3) 由于比height[i]大的元素都出完了，height[i]又比栈顶元素大了，因此再次进栈。如此往复，直到遍历到最后那个高度为0的柱，触发最后的弹出以及最后一次面积的计算， 此后stack为空。
(4) 返回面积最大值。
栈中存的不是高度，而是height的索引，这样做的好处是不会影响宽度的计算，索引值相减 = 宽度。
虽然是二重循环，但时间复杂度实际  2N，故为O(N)

我的代码：

// 88ms
class Solution {
public:
    int largestRectangleArea(vector<int> &height) {
        int sum = 0, max_sum = 0, tmp = 0;
        if (height.size() == 0) return 0;
        height.push_back(0);
        stack<int> stack;
        for (int i = 0; i < height.size(); ++i) {
            if (stack.empty() || height[i] > height[stack.top()]) {
                //std::cout<<"push i = "<<i<<std::endl;
                stack.push(i);
            } else {
                tmp = stack.top();
                stack.pop();
//                sum = height[tmp] * (i - tmp); //会错误计算={2,1,5,6,2,3};中1,2,3的stack构成的矩形。
                sum = height[tmp] * (stack.empty()? i : (i - stack.top() - 1));
                max_sum = max_sum > sum ? max_sum : sum;
//                std::cout<<"sum = "<<sum<<", i = "<<i<<", height[tmp] = "<<height[tmp]<<", tmp = "<<tmp<<std::endl;
                i--;
            }
        }
        return max_sum;
    }
};

参考资料：

1 felixfang's blog