单调栈主要用来找一个元素左边第一个比它小(大)的元素
例如我们需要查找每一个元素向左第一个比它小的元素,我们只要维护一个单调增的序列(从栈底到栈顶,下标依次增大,值也依次增大)。
因为对于任意的i,j,若i>j且a[i]<a[j],那么找第一个小于a[k]的答案时,必然不会选到a[j]。
我们淘汰掉当前栈中大于等于即将入栈的元素,这样当前的栈顶就是我们需要的答案,最后再把当前元素入栈。
for (int i = 1; i <= n; i++)
{
while( !s.empty() && a[i] < a[s.top()] ) s.pop();
if( s.empty() ) l[i] = 0; //l[i]为第一个小于等于a[i]的元素下标
else l[i] = s.top();
s.push(i);
}
hdu1506
1-n位置,每个位置上都有一个宽度为1高度为a[i]的矩形,要求在这样的立方图中找到一个矩形,使得矩形面积尽可能大。
分析:我们利用单调栈找到一个元素左边第一个比它小的元素,那么这个小的元素位置就是这个高度扩展的左边界,那么对于一个即将被弹出去的元素,这个元素就找到了右边第一个小于它的位置,这时左边界在栈顶的下一个元素,这样这个元素的左右边界都找到了,这时候就直接算这个元素的面积。由于并不是所有元素都能被弹出,所以要额外在最后加一个0。
#include <iostream>
#include <stack>
using namespace std;
typedef long long ll;
const int maxn = 1e5+5;
ll a[maxn];
stack<int> s;
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n;
while( cin >> n && n != 0 )
{
ll ans = 0;
for (int i = 1; i <= n; i++) cin >> a[i];
a[n+1] = 0; //额外加一个0元素
while( !s.empty() ) s.pop(); //清空栈
for (int i = 1; i <= n + 1; i++)
{
while( !s.empty() && a[s.top()] > a[i] ) //栈中元素比当前的大,这个元素的右边界找到了
{
int x = s.top();
s.pop();
int len = i - x - 1; //计算右边扩展的宽度
if( s.empty() ) len += x; //左边第一个比它小的元素就在栈的第二个位置,弹出后现在在栈顶
else len += x - s.top();
ans = max(ans,len*a[x]);
}
s.push(i);
}
cout << ans << '\n';
}
return 0;
}