单调栈时,将每个值作为最小值寻找其左右最长能够延伸的位置,再利用前缀和数组进行更新最大值
然后处理负数时,我们总是尽可能希望负数能够和负数相乘,这样才能最大化
#include<iostream>
#include<cstdio>
#include<stack>
#include<cstring>
using namespace std;
const int maxn=5e5+5;
typedef long long LL;
typedef pair<LL,int>PII;
LL max(LL a,LL b)
{
return a>b?a:b;
}
int main()
{
int n;
LL a[maxn],sum[maxn],b[maxn],c[maxn],sum1[maxn];
scanf("%d",&n);
sum[0]=sum1[0]=0;
int left[maxn],right[maxn];
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
}
memcpy(b,a,sizeof(a));
for(int i=1;i<=n;i++)
{
if(b[i]<0) b[i]=0;
sum[i]=1l*sum[i-1]+b[i]; //前缀和
left[i]=right[i]=i;
}
b[n+1]=0;
stack<PII>str;
while(!str.empty()) str.pop();
//单调栈,以每一个数作为延伸,记录延伸范围
for(int i=1;i<=n+1;i++)
{
if(str.empty()||b[i]>str.top().first) {
str.push(PII(b[i],i));
}
else {
while(!str.empty()&&b[i]<=str.top().first)
{
PII x=str.top();
str.pop();
left[i]=left[x.second];
right[x.second]=i-1;
}
str.push(PII(b[i],i));
}
}
LL solve=-1;
for(int i=1;i<=n;i++)
{
if(1l*b[i]*(sum[right[i]]-sum[left[i]-1])>solve) {
solve=1l*b[i]*(sum[right[i]]-sum[left[i]-1]);
}
}
while(!str.empty())
{
str.pop();
}
for(int i=1;i<=n;i++)
c[i]=-a[i];
LL cur=0,mx=0;
for(int i=1;i<=n;i++)
{
if(cur<0) cur=0,mx=0;
cur+=c[i];
mx=max(mx,c[i]);
solve=max(solve,cur*mx);
}
cout<<solve<<endl;
return 0;
}