以下先贴代码
#include<cstdio>
#include<stack>
using namespace std;
inline int read(){
int t=getchar(), p=0;
while(t<'0'||t>'9') t=getchar();
while(t>='0'&&t<='9') p=p*10+t-48, t=getchar();
return p;
}
inline void ma(long long& x, long long y){
if(x<y) x=y;
}
stack<int>s;
int a[50111], x[50111], y[50111];
int main(){
int n=read();
a[0]=a[n+1]=0;
for(int i=1;i<=n;i++) a[i]=read();
s.push(0);
for(int i=1;i<=n;i++){
while(a[i]<=a[s.top()]) s.pop();
x[i]=i-s.top();
s.push(i);
}
while(!s.empty()) s.pop();
s.push(n+1);
for(int i=n;i>0;i--){
while(a[i]<=a[s.top()]) s.pop();
y[i]=s.top()-i;
s.push(i);
}
long long ans=0;
for(int i=1;i<=n;i++) ma(ans,(long long)a[i]*(x[i]+y[i]-1));//50000*1e9 会溢出,要用long long
printf("%lld\n", ans);
return 0;
}
主要思路是单调栈
因为选出的矩形的高度一定是某一个位置的高,所以可以预处理出以每个位置为高可以向左(x[i])和向右(y[i])拓展的个数,然后ans=ma(ans,a[i]*(x[i]+y[i]-1))这样就可以得到答案,复杂度为n^2
然而拓展的个数可以借单调栈实现,由此复杂度降为O(N)
以向左为例,单调栈维护一个单调递增序列,求得在每个点左边第一个比i点小的数
如样例
6 2 1 5 6 2 3
为了方便,虚拟一个a[0]=0,这样就不会RE
i=1时,2>0,入栈 , 栈内为{0,2} 第一个比2小的是第0位的0 x[1]=1-0=1;
i=2时,1<2,2出栈,1入栈, 栈内为{0,1}第一个比1小的是第0位的0 x[2]=2-0=2;
i=3时,5>3, 5入栈 栈内为{0,1,5}第一个比5小的是第二位的1 x[3]=3-2=1
i=4时,6>5, 6入栈 栈内为{0,1,5,6}第一个比6小的是第三位的5 x[4]=4-3=1
i=5时,2<6, 6出栈,2<5, 5出栈, 2入栈 栈内为{0,1,2}第一个比2小的是第2位的1 x[5]=5-2=3
i=6时,3>2,3入栈 栈内为{0,1,2,3}第一个比3小的是第5位的2 x[6]=6-5=1
y[i]同理可以推出来