poj2796 Feel Good (单调栈+前缀和)

Feel Good

Bill is developing a new mathematical theory for human emotions. His recent investigations are dedicated to studying how good or bad days influent people’s memories about some period of life.

A new idea Bill has recently developed assigns a non-negative integer value to each day of human life.

Bill calls this value the emotional value of the day. The greater the emotional value is, the better the daywas. Bill suggests that the value of some period of human life is proportional to the sum of the emotional values of the days in the given period, multiplied by the smallest emotional value of the day in it. This schema reflects that good on average period can be greatly spoiled by one very bad day.

Now Bill is planning to investigate his own life and find the period of his life that had the greatest value. Help him to do so.

Input

The first line of the input contains n - the number of days of Bill’s life he is planning to investigate(1 <= n <= 100 000). The rest of the file contains n integer numbers a1, a2, … an ranging from 0 to 10 6 - the emotional values of the days. Numbers are separated by spaces and/or line breaks.

Output

Print the greatest value of some period of Bill’s life in the first line. And on the second line print two numbers l and r such that the period from l-th to r-th day of Bill’s life(inclusive) has the greatest possible value. If there are multiple periods with the greatest possible value,then print any one of them.

Sample Input

6
3 1 6 4 5 2

Sample Output

60
3 5

分析:

单调栈求出每个数左右两边能拓展的最大距离
前缀和用来求区间和

code1：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
//#define ll long long
#define int long long
using namespace std;
const int maxm=1e5+5;
int a[maxm];
int l[maxm],r[maxm];
int sum[maxm];
int stk[maxm];
int n;
int L,R;
signed main(){
    scanf("%I64d",&n);
    for(int i=1;i<=n;i++){
        scanf("%I64d",&a[i]);
        sum[i]=sum[i-1]+a[i];
    }
    a[0]=a[n+1]=-1;
    int head=0;
    for(int i=1;i<=n+1;i++){
        while(head&&a[i]<a[stk[head]]){
            r[stk[head]]=i-1;
            head--;
        }
        stk[++head]=i;
    }
    head=0;
    for(int i=n;i>=0;i--){
        while(head&&a[i]<a[stk[head]]){
            l[stk[head]]=i+1;
            head--;
        }
        stk[++head]=i;
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        int temp=(sum[r[i]]-sum[l[i]-1])*a[i];
        if(temp>=ans){
            ans=temp;
            L=l[i];
            R=r[i];
        }
    }
    printf("%I64d\n",ans);
    printf("%I64d %I64d\n",L,R);
    return 0;
}

code2：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
//#define ll long long
#define int long long
using namespace std;
const int maxm=1e5+5;
int a[maxm];
int l[maxm],r[maxm];
int sum[maxm];
int stk[maxm];
int n;
int L,R;
signed main(){
    scanf("%I64d",&n);
    for(int i=1;i<=n;i++){
        scanf("%I64d",&a[i]);
        sum[i]=sum[i-1]+a[i];
    }
    a[n+1]=-1;
    int head=0;
    int ans=-1;
    for(int i=1;i<=n+1;i++){
        if(head==0||a[i]>=a[stk[head]]){
            stk[++head]=i;
        }else{
            int last=i;
            while(head&&a[i]<a[stk[head]]){
                last=stk[head];
                head--;
                int temp=(sum[i-1]-sum[last-1])*a[last];
                if(temp>ans){
                    ans=temp;
                    L=last;
                    R=i-1;
                }
            }
            stk[++head]=last;
            a[last]=a[i];
        }
    }
    printf("%I64d\n",ans);
    printf("%I64d %I64d\n",L,R);
    return 0;
}