南昌网络赛 I. Max answer 单调栈

Max answer

题目链接

https://nanti.jisuanke.com/t/38228

Describe

Alice has a magic array. She suggests that the value of a interval is equal to the sum of the values in the interval, multiplied by the smallest value in the interval.

Now she is planning to find the max value of the intervals in her array. Can you help her?

Input

First line contains an integer n(1≤n≤5*10^5)

Second line contains n integers represent the array a(−10^5≤ai≤10^5)

Output

One line contains an integer represent the answer of the array.

样例输入

5

1 2 3 4 5

样例输出

36

题意
给你一个序列，对于每个连续子区间，有一个价值，等与这个区间和×区间最小值，
求所有子区间的最大价值是多少。

题解

a[i]表示第i位的权值，L[i]表示第i个数左边第一个比他小的数，R[i]表示右边第一个比它小的数。

我们可以用单调站求出来，不会可以先做洛谷的完美序列,　https://www.luogu.org/problemnew/show/P2659

然后对于每一位i，如果a[i]为正数，我们可以求包含i的区间和最大的子区间，然后再乘上这个值，当然不能越过L[i]和R[i]。

我们怎么求呢？

我们先定义sum(i,j)为i到j的区间和，再定义lmax[i]为以i 为右端点的使得区间和和最大的左端点位置，(即求一个l<=i,使得sum(l,i)最大)。

那么，lmax[i]可以通过lmax[i-1]更新，这个就是一个贪心吧，分两种情况：

１．sum(lmax[i-1],i-1)<0, 那么lmax[i]=i.

２．sum(lmax[i-1],i-1)>0,lmax[i+1]=lmax[i].

相似的我们可以求出lmax[i],lmin[i],rmax[i]。对于i ,若a[i]>0，其最大贡献就是sum(max(L[i]+1,lmax[i]),min(R[i]-1,rmax[i])*a[i]。负数类似。

这样题目就做完啦，时间复杂度O(n)

代码：

#include<bits/stdc++.h>
using namespace std;
#define N 3000050
#define INF 0x7f7f7f7f
#define ll long long
template<typename T>void read(T&x)
ll n,a[N],s[N],sk[N],L[N],R[N];
ll Rmin[N],Lmin[N],Rmax[N],Lmax[N],ans,top;
int main()
{
#ifndef ONLINE_JUDGE
  freopen("aa.in","r",stdin);
#endif
  ios::sync_with_stdio(false);
  cin>>n;
  for(ll i=1;i<=n;i++)cin>>a[i],s[i]=a[i]+s[i-1];
  ans=a[1]*a[1];
  sk[top=1]=0;
  a[0]=-INF-1;
  a[n+1]=-INF;
  for(ll i=1;i<=n+1;i++)
    {
      while(a[i]<a[sk[top]])
    R[sk[top--]]=i;
      L[i]=sk[top];
      sk[++top]=i;
    }
  Lmin[1]=1;
  Lmax[1]=1;
  Rmin[n]=n;
  Rmax[n]=n;
  for(ll i=2;i<=n;i++)
    {
      if (s[i-1]-s[Lmin[i-1]-1]>0)Lmin[i]=i;
      else Lmin[i]=Lmin[i-1];
      if (s[i-1]-s[Lmax[i-1]-1]<0)Lmax[i]=i;
      else Lmax[i]=Lmax[i-1];
    }
  for(ll i=n-1;i>=1;i--)
    {
      if (s[Rmin[i+1]]-s[i]>0)Rmin[i]=i;
      else Rmin[i]=Rmin[i+1];
      if (s[Rmax[i+1]]-s[i]<0)Rmax[i]=i;
      else Rmax[i]=Rmax[i+1];
    }
  ll l,r;
  for(ll i=1;i<=n;i++)
    {
      if (a[i]<0)
    {
      l=max(L[i]+1,Lmin[i]);
      r=min(R[i]-1,Rmin[i]);
    }
      else
    {
      l=max(L[i]+1,Lmax[i]);
      r=min(R[i]-1,Rmax[i]);
    }
      ans=max((s[r]-s[l-1])*a[i],ans);
    }
  cout<<ans;
}