题意
给一个直方图,求直方图中的最大矩形的面积。例如,下面这个图片中直方图的高度从左到右分别是2, 1, 4, 5, 1, 3, 3, 他们的宽都是1,其中最大的矩形是阴影部分。
Input
输入包含多组数据。每组数据用一个整数n来表示直方图中小矩形的个数,你可以假定1 <= n <= 100000. 然后接下来n个整数h1, …, hn, 满足 0 <= hi <= 1000000000. 这些数字表示直方图中从左到右每个小矩形的高度,每个小矩形的宽度为1。 测试数据以0结尾。
Output
对于每组测试数据输出一行一个整数表示答案。
Sample Input
7 2 1 4 5 1 3 3
4 1000 1000 1000 1000
0
Sample Output
8
4000
单调栈理解
单调栈=单调+栈;
单调性: 只可意会不可言传 擅用搜索引擎 ;
栈:先进后出的一种线性数据结构(更多内容参考搜索引擎);
单调栈,有序的栈(一定是从栈顶到栈底满足单调性),push过程中删除破坏有序结构的数据(菜鸡理解 );
首先来看一下单调栈的性质:
单调栈的类型:
单调递增栈(从栈顶到栈底满足递增)
单调递减栈(从栈顶到栈底满足递减)
单调栈的作用:
线性的时间复杂度:每个数据只入栈一次,出栈一次,所以O(n)的时间复杂度
单调递增栈:可以找到从当前元素往左/往右第一个比当前元素大的元素;
单调递减栈:可以找到从当前元素往左/往右第一个比当前元素小的元素;
可以求得以当前元素为最值得最大区间(结合上面两条自行理解)。
用单调递减栈举个栗子(递增栈类似):
有一组数据:
2 1 4 5 1 3 3
以第一个元素2
为栗,右边第一个比2小的元素为1(下标为2)
,左边没有,所以为0;
以中间元素5为栗,左边第一个比5小的元素为4
,右边第一个比5小的元素为1(下标为5)
。
类似的,可以求出其他元素的往左/往右的第一个小于当前元素的值。
到这我们再看数列:
我们发现,被弹出元素右边第一个比自己小的元素就是弹出自己的元素。
反过来入栈,我们即可求出左边比自己小的元素。
思路
最大的矩形面积,我们只需要找到当前高度可以延展的最大宽度即可。
用题目中的数据举例:
1, 4, 5, 1, 3, 3
1可以向左向右延展的最大区间为[1,6],所以以1为高度,面积为6×1=6;
4可以向左向右延展的最大区间为[2,3],因为1和4都比它低 ,不能延展出去,换句话说,只有两边比它高的才能延展出去,所以以4为高度,面积为4×2=8;
…(剩下的自己求一下)
最后找出最大面积即可。
代码:
#include<iostream>
#include <cstdio>
using namespace std;
int n;
long long l[100010]= {0};
long long r[100010]= {0};
long long a[100010];
long long st[100010]; //区分相同元素 所以存储下标 数组模拟栈
void getRight() //得到每一个右边元素比自己小的元素下标
{//模拟单调栈的规则
int t=0;
for(int i=n-1; i>=0; i--)
{
while(t>0&&a[i]<=a[st[t-1]])
{
t--;
}
if(t==0) //说明有边界为栈底 可以延展到最后
r[i]=n;
else
r[i]=st[t-1];
st[t++]=i;
}
}
void getLeft() //得到每一个左边元素比自己小的元素下标
{
int t=0;
for(int i=0; i<n; i++)
{
while(t>0&&a[i]<=a[st[t-1]])
{
t--;
}
if(t==0)
l[i]=0; //说明有边界为栈顶 可以延展到最开始
else
l[i]=st[t-1]+1;
st[t++]=i;
}
}
int main()
{
while(1)
{
scanf("%d",&n);
if(n==0) break;
for(int i=0; i<n; i++)
{
scanf("%lld",&a[i]);
}
getLeft();
getRight();
long long ans = 0;
for (int i = 0; i <n; i++)
{
ans=max(ans,(r[i] - l[i] ) * a[i]);
}
cout<<ans<<endl;
}
return 0;
}
我们进一步探讨:
2 1 4 5 1 3 3
这组数据有7个数,刚才我们只找到了3个往右的值。
现在我们假设,在数据最开始和末尾各放一个零,那么我们是不是就可以找到所有元素的往左/往右的第一个比它小的值,即找到0,就意味着可以往左/往右没有比它小的元素。
到这
我们发现,一个元素,被谁弹出来,谁就是其右边的第一个比它小的元素,
这个元素在栈内右边的元素,是其左边的第一个比它小的元素。
(如果有两个相同元素,则只有一个范围正确,第二个范围 有点偏差 )
(上图使用条件是栈内不严格单调,使用严格单调则正确的那个值会返回反过来,总之只有一个正确,也可能是我的推导错了,欢迎指正 )
用这个方法,我们只需遍历一边即可拿到左右边界值。
代码
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include<stack>
#include<algorithm>
#define ll long long
using namespace std;
ll a[100005];
ll r[100005];
ll l[100005];
int main()
{
int n;
while (1)
{
scanf("%d", &n);
if (n == 0) return 0;
for (int i = 1; i <= n; i++)
{
scanf("%lld", &a[i]);
}
a[n + 1] = 0; //最后一个0
stack<int> st;
st.push(0); //第一个0
for (int i = 1; i <= n + 1; i++)
{
while (!st.empty() && a[i] < a[st.top()])
{
r[st.top()] = i;
//被谁弹出来,谁就是其右边的第一个比它小的元素
st.pop();
}
if(!st.empty()) l[i] = st.top(); //入栈前的栈顶是他的右边元素
//栈内右边的元素,是其左边的第一个比它小的元素
st.push(i);
}
long long ans = 0;
for (int i = 1; i <= n; i++)
{
ans = max(ans, (r[i] - l[i]-1) * a[i]);
}
cout << ans << endl;
}
return 0;
}
总结:
这道题得开long long 要不然过不去。
一定要先用手过一遍,看看栈内是严格单调还是不严格单调。