leetcode 随笔 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.

 

Example:

Input: [2,1,5,6,2,3]
Output: 10

这道题的标准解法是维护一个递增栈，以[1,2,3,4,5,3,4,5,4,5]为例

如果输入>=栈顶，则进栈，否则出栈，直到栈顶<输入或者栈为空。

最后栈的结果应该是[1,2,3,3,4,4,5]

在这里我们在栈中存放的是输入的位置，这样就可以根据当前输入的值求最大矩形的面积了。

每次出栈时都要计算一下当前可能的最大矩形面积的大小。

如果出栈后栈为空说明当前值是所有最小值，应该是输入的个数*当前的高度。

· 可以理解一下[5,4,4,2,4,3,3]的情况

开始0（0为输入的位置，值为5）进栈

然后是0出栈，1进栈，（5>4，不满足递增序列）,判断当前最大值为1*5

2进栈

1,2出栈，3进栈，判断当前最大值为 3*4 (输入的个数3，高度4)

4进栈

4出栈，5进栈，判断最大值

6进栈，判断最大值

此时输入结束，栈内元素为[3,5,6]

栈依次出栈判断最大值为 7*2（7为输入的个数*2为栈最后一个元素的高度）

class Solution {
public:
	int largestRectangleArea(vector<int>& heights) {
		if (heights.empty()) return 0;
		int mx = INT_MIN;
		int i = 0;
		stack<int> record;
		for (; i < heights.size(); i++)
		{
			while (!record.empty() && heights[record.top()] > heights[i])
			{
				int c = heights[record.top()];
				record.pop();
				c = record.empty() ? i*c : (i - record.top() - 1)*c;
				mx = c > mx ? c : mx;
			}
			record.push(i);
		}
		while (!record.empty())
		{//i==heights.size()
			int c = heights[record.top()];
			record.pop();
			c = record.empty() ? i*c : (i - record.top() - 1)*c;
			mx = c > mx ? c : mx;
		}
		return mx;
	}
};