LeetCode | Largest Rectangle in Histogram

来源:互联网 发布:计算机应用与软件 编辑:程序博客网 时间:2024/06/01 10:19

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.

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

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

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

自然大家都知道O(n^2)的解法,也就是两个for循环,分别表示区间的起点和终点,可以在O(n^2)的时间内得出结果。

而利用栈可以在O(n)的时间复杂度内完成。
具体可参照这篇博文:http://blog.csdn.net/doc_sgl/article/details/11805519
讲的十分细致和清楚。

首先使用的栈存储的是一段连续递增的下标。
当遇到比栈顶元素大的元素,进栈;否则,出栈并计算当前区间所构成的面积大小。
面积的计算公式为:
int t=Q.top();Q.pop();
heights[t]*(Q.empty()?i:i-Q.top()-1);

一开始我所写为计算完了area再pop(),所以公式是heights[t]*(i-Q.top());
但是这样计算是不完整的,因为,例如对于[2,1,2]
按照我的办法求出的是如下区间
3

但是实际上我们知道,栈顶的最后一个元素是i元素之前最小的元素。
并且因为栈内存的一直是以递增元素排序的下标,可以保证先弹出的元素全部比当前栈顶元素小。
于是上图的黄色区域可以继续向左扩展。
4
也即可以扩展到i之前的所有元素。
故需要先弹出栈顶元素,并对栈空作一个判定

class Solution {public:    int largestRectangleArea(vector<int>& heights) {        heights.push_back(0);        stack<int> Q;        int i=0,maxV=0;        int n=heights.size();        while(i<n){            if(Q.empty() || heights[i]>=heights[Q.top()])            Q.push(i++);            else{                int h=Q.top();                Q.pop();                //这里的判断非常关键,如果是栈内最后一个元素,则不应当计算一段区间而应当计算从0开始所有的高度                int area=(Q.empty()?i:i-Q.top()-1)*heights[h];                maxV=max(maxV,area);            }        }        return maxV;    }};
0 0