bzoj3745: [Coci2015]Norma

Description

Input

第1行，一个整数N；

第2~n+1行，每行一个整数表示序列a。

Output

输出答案对10^9取模后的结果。

预处理每个位置的数作为最小/大值向左延伸的最大距离，线段树维护序列的前缀的后缀min和后缀max以及这个前缀的后缀对答案的贡献，在前缀末尾加入一个数可以快速维护。

#include<cstdio>
typedef long long i64;
const int N=,P=;
char buf[N*],*ptr=buf-;
int _(){
    int x=,c=*++ptr;
    while(c<)c=*++ptr;
    while(c>)x=x*+c-,c=*++ptr;
    return x;
}
int _l,_r,_a,v1,v2,ans=;
inline int mod(int x){return x<P?x:x-P;}
struct node{
    node*l,*r;
    int f1,f2,sz,L,R;
    int s1,s2,s3,ss1,ss2,ss3;
    void fil1(int x){
        f1=x;
        s1=i64(x)*sz%P;
        ss1=(i64(sz)*(sz+)>>)%P*x%P;
        s3=i64(x)*s2%P;
        ss3=i64(x)*ss2%P;
    }
    void fil2(int x){
        f2=x;
        s2=i64(x)*sz%P;
        ss2=(i64(sz)*(sz+)>>)%P*x%P;
        s3=i64(x)*s1%P;
        ss3=i64(x)*ss1%P;
    }
    void dn(){
        if(f1)l->fil1(f1),r->fil1(f1),f1=;
        if(f2)l->fil2(f2),r->fil2(f2),f2=;
    }
    void up(){
        s1=mod(l->s1+r->s1);
        s2=mod(l->s2+r->s2);
        s3=mod(l->s3+r->s3);
        ss1=(l->ss1+r->ss1+i64(l->s1)*r->sz)%P;
        ss2=(l->ss2+r->ss2+i64(l->s2)*r->sz)%P;
        ss3=(l->ss3+r->ss3+i64(l->s3)*r->sz)%P;
    }
    void set1(){
        if(_l<=L&&R<=_r){
            fil1(_a);
            return;
        }
        dn();
        int M=L+R>>;
        if(_l<=M)l->set1();
        if(_r>M)r->set1();
        up();
    }
    void set2(){
        if(_l<=L&&R<=_r){
            fil2(_a);
            return;
        }
        dn();
        int M=L+R>>;
        if(_l<=M)l->set2();
        if(_r>M)r->set2();
        up();
    }
    void get(){
        if(R<=_r){
            v2=(v2+ss3+i64(v1)*sz)%P;
            v1=mod(v1+s3);
            return;
        }
        dn();
        int M=L+R>>;
        l->get();
        if(_r>M)r->get();
    }
}ns[N*],*np=ns,*rt;
node*build(int L,int R){
    node*w=np++;
    w->L=L;w->R=R;
    w->sz=R-L+;
    if(L!=R){
        int M=L+R>>;
        w->l=build(L,M);
        w->r=build(M+,R);
    }
    return w;
}
int n,a[N],p1[N],p2[N],ss[N],sp=;
int main(){
    fread(buf,,sizeof(buf),stdin);
    n=_();
    for(int i=;i<=n;++i)a[i]=_();
    for(int i=n;i;--i){
        while(sp&&a[ss[sp]]>a[i])p1[ss[sp--]]=i+;
        ss[++sp]=i;
    }
    while(sp)p1[ss[sp--]]=;
    for(int i=n;i;--i){
        while(sp&&a[ss[sp]]<a[i])p2[ss[sp--]]=i+;
        ss[++sp]=i;
    }
    while(sp)p2[ss[sp--]]=;
    rt=build(,n);
    for(int i=;i<=n;++i){
        _l=p1[i];_r=i;_a=a[i];
        rt->set1();
        _l=p2[i];
        rt->set2();
        v1=v2=;
        rt->get();
        ans=mod(ans+v2);
    }
    printf("%d",ans);
    return ;
}