POJ 2796 Feel Good（单调栈）

Description
给出一个长度为n的序列，定义一个区间的参考值：区间和*区间最小值，求这个序列中的最大区间参考值以及其对应的区间
Input
第一行为一整数n表示序列长度，第二行n个整数表示这个序列(1<=n<=100 000)
Output
Sample Input
6
3 1 6 4 5 2
Sample Output
60
3 5
Solution
首先求个前缀和sum[i]，然后用两遍单调栈求出每个元素ai是最小值的最大区间[li,ri]，那么ans=max(ai*(sum[ri]-sum[li-1]))
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
#define maxn 111111
typedef long long ll;
int n,a[maxn],sta[maxn],p,l[maxn],r[maxn],ansl,ansr;
ll sum[maxn],ans;
int main()
{
    while(~scanf("%d",&n))
    {
        sum[0]=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            sum[i]=a[i]+sum[i-1];
            l[i]=r[i]=i;
        }
        p=0;
        for(int i=1;i<=n;i++)
        {
            if(!p||a[i]>=a[sta[p]])sta[++p]=i;
            else 
            {
                while(p&&a[i]<a[sta[p]])
                    r[sta[p]]=i-1,p--;
                sta[++p]=i;
            }
        }
        while(p)r[sta[p]]=n,p--;
        for(int i=n;i>=1;i--)
        {
            if(!p||a[i]>=a[sta[p]])sta[++p]=i;
            else 
            {
                while(p&&a[i]<a[sta[p]])
                    l[sta[p]]=i+1,p--;
                sta[++p]=i;
            }
        }
        while(p)l[sta[p]]=1,p--;
        ans=0;
        for(int i=1;i<=n;i++)
            if(ans<=1ll*a[i]*(sum[r[i]]-sum[l[i]-1]))
                ans=1ll*a[i]*(sum[r[i]]-sum[l[i]-1]),ansl=l[i],ansr=r[i];
        printf("%lld\n%d %d\n",ans,ansl,ansr); 
    }
    return 0;
}