这里维护一个单调递减栈,首先如果栈空或者比栈顶元素大的数的下标直接加入栈中,如果与栈顶元素相等可以不用处理, 如果比栈顶元素小,那么需要不断更新栈顶元素,并且更新面积,一直到栈顶元素小于当前元素为止。把f[n]=-1,这样可以把栈中所有元素算完。
比如样例 2,1,5,6,2,3 首先将2入栈,因为1比2小,那么就把栈中元素的最大面积算出来,就是2,并且弹栈,保存退栈的元素下标,就是0,然后赋值f[0]=1,因为1可以延伸到第一个,那么加入栈中的下标是0,并且f[0]=1,遇到5,6都是直接加入栈中,遇到2,当前栈顶元素是6,所以算出6这点的面积,并且保存下标,此时栈顶元素是5,那么算出面积是10,sum = (i-q.top())*f[q.top()]; 因为i跟q.top隔了2,然后保存2的下标并且赋值为2,表示2这点可以向前延伸的最长距离,遇到3直接加入,然后遇到-1,算出每一个点能表示的最大面积。
#include<iostream>
#include<stack>
#include<algorithm>
#include<cmath>
using namespace std;
long long max(long long a,long long b)
{
return a>b?a:b;
}
int main()
{
int n;cin>>n;
long long ans=0,a[50000+5];
for(int i=0;i<n;i++) cin>>a[i];
stack<int>s;
while(!s.empty()) s.pop();
a[n]=-1;
for(int i=0;i<=n;i++)
{
if(s.empty()||a[i]>a[s.top()])
{
s.push(i);
}
else if(a[i]<a[s.top()])
{
int t=i;
while(!s.empty()&&a[i]<a[s.top()])
{
ans=max(ans,(i-s.top())*a[s.top()]);
t=s.top();
s.pop();
}
s.push(t);
a[t]=a[i];
}
}
cout<<ans;
return 0;
}