GDUT专题学习5- A - Largest Rectangle in a Histogram

A - Largest Rectangle in a Histogram

直方图是一个多边形，由一系列在公共基线上对齐的矩形组成。矩形具有相等的宽度，但是可以具有不同的高度。例如，左图显示了由高度为2、1、4、5、1、3、3的矩形组成的直方图，以1为单位测量，其中1是矩形的宽度:

通常，直方图用于表示离散分布，例如文本中字符的频率。注意，矩形的顺序，即它们的高度，是很重要的。计算直方图中与公共基线对齐的最大矩形的面积。右图显示了所描绘的直方图的最大对齐矩形。

输入

输入包含几个测试用例。每个测试用例描述一个直方图，并以一个整数开始n，表示它由多少个矩形组成。你可以假设1<=n<=100000。那就跟着n整数h1,...，hn，在哪里0<=hi<=1000000000。这些数字以从左到右的顺序表示直方图矩形的高度。每个矩形的宽度为1。最后一个测试用例的输入后面跟一个零。

输出

对于单行上输出的每个测试用例，指定直方图中最大矩形的面积。记住，这个矩形必须在公共基线上对齐。

样例

输入	输出
7 2 1 4 5 1 3 3 4 1000 1000 1000 1000 0	8 4000

思路

题目大意是求最大矩形面积，所以可以以单个矩形的高为起点进行左右延伸找到最大延伸距离，算出每个矩形对应面积，维护一个最大值即可。纯暴力会超时，这里利用单调栈进行优化。

代码

#include<iostream>
#include<stack>
#define LL long long
using namespace std;
const int N=1e5+10;
LL a[N],l[N],r[N];
int main(){
	int n;
	LL ans;
	while(cin>>n&&n){
		stack<int>st;
		ans=0;
		for(int i=1;i<=n;i++)cin>>a[i];
		for(int i=1;i<=n;i++){
			while(!st.empty()&&a[i]<=a[st.top()]){
				st.pop();
			}
			if(st.empty()) l[i]=1;
			else l[i]=st.top()+1;
			st.push(i);
		}
		while(!st.empty()) st.pop();
		for(int i=n;i>=1;i--){
			while(!st.empty()&&a[i]<=a[st.top()]){
				st.pop();
			}
			if(st.empty()) r[i]=n+1;
			else r[i]=st.top();
			st.push(i);
		}
		for(int i=1;i<=n;i++){
			ans=max(ans,a[i]*(r[i]-l[i]));
		}
		cout<<ans<<endl;
	}
	return 0;
}