【单调栈】最大矩形

问题描述

给一个直方图，求直方图中的最大矩形的面积。例如，下面这个图片中直方图的高度从左到右分别是2, 1, 4, 5, 1, 3, 3, 他们的宽都是1，其中最大的矩形是阴影部分。

Input&Output

• Input

输入包含多组数据。每组数据用一个整数n来表示直方图中小矩形的个数，你可以假定1 <= n <= 100000. 然后接下来n个整数h1, …, hn, 满足 0 <= hi <= 1000000000. 这些数字表示直方图中从左到右每个小矩形的高度，每个小矩形的宽度为1。 测试数据以0结尾。

• Output
  对于每组测试数据输出一行一个整数表示答案。

Sample

Sample Input

7 2 1 4 5 1 3 3
4 1000 1000 1000 1000
0

Sample Output

8
4000

思路

根据数据范围，复杂度允许$O（nlogm)$以下.所以我们来考虑哪些矩形对最优解有贡献。若h_i>h_i+1，那么hi对于以后的组合是没有贡献的。所以h_i高度有效的范围是ij(h_j是[i,j]中最后一个大于等于h_i的高度)。所以我们要求每个以l为左边界的最大右边界r，其面积(r-l+1)*h_l作为最大面积的贡献之一。多次模拟过程发现，维护最大右边界的过程符合单调增栈结构.规则如下：

• 若栈为空或h_i大于栈顶元素，{i,h_i}入栈。
• 否则，弹出所有小于等于h_i的元素h_j，并且计算以h_j为高，i-j+1为宽的最大面积。直到栈为空或栈顶元素小于h_i，{i,h_i}入栈。
• 循环结束后，要结算栈中所有元素，此时右边界为n。

每个元素最多入栈出栈各一次，复杂度为$O(n)$。

Tips

注意数值范围。
longlong*int=longlong，所以不用全开longlong。序号用int，值用longlong。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iomanip>
#include<stack>
using namespace std;
typedef long long LL; 
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define mem(a,s) memset(a,s,sizeof(a))
LL R(){//读入正数(开longlong) 
	LL w=0;char c=getchar();
	while(c>='0'&&c<='9'){
		w=w*10+(c-'0');
		c=getchar();
	}
	return w;
}
struct node{int v;LL h;};
stack<node> st;
int main(){
//	freopen("1.txt","r",stdin);
	int n;
	while(true){
		n=R(); 
		if(n==0)break;
		LL max=0,m;
		rep(i,1,n+1){
			if(i<=n) 
				m=R();
			else m=0;//最后一次弹出所有元素 
			int l=i;
			while(!st.empty()&&st.top().h>=m){
				l=st.top().v;
				LL si=(i-st.top().v)*st.top().h;
				max=(max>si?max:si);
				st.pop();
			}
			if(i<=n){
				node x;x.h=m;x.v=l;
				st.push(x);			
			}
		}
		cout<<max<<endl;
	} 
	return 0;
}