【jzoj5231】【NOIP2017模拟A组模拟8.5】【序列问题】 【分治】

题目大意

解题思路

考虑分治，统计跨国分治中心的区间的答案，从左到右枚举右端点，维护第一个左端点到分治中心max比分治中心到右端点大的位置，同理维护min，同时维护对答案的贡献即可。

code

#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define ULL unsigned int
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define fr(i,j) for(int i=begin[j];i;i=next[i])
using namespace std;
int const mn=5*1e5+9,inf=1e9+7,mo=1e9+7;
int n,ans,a[mn],f[mn],g[mn];
int max(int x,int y){return (x>y)?x:y;}
int min(int x,int y){return (x<y)?x:y;}
void solve(int l,int r){
    if(l>=r){
        if(l==r)ans=(ans+1ll*a[l]*a[l])%mo;
        return;
    }
    int md=(l+r)/2;
    solve(l,md);
    solve(md+1,r);
    int mx=0,mi=inf,j=md,k=md,sum=0,su2=0;
    f[md+1]=0;g[md+1]=inf;
    fd(i,md,l)f[i]=max(f[i+1],a[i]),g[i]=min(g[i+1],a[i]),
        sum=(sum+1ll*f[i]*g[i])%mo;
    int tmp=ans;
    fo(i,md+1,r){
        mx=max(mx,a[i]);
        mi=min(mi,a[i]);
        while((j>=l)&&(f[j]<mx)){
            if(j<=k)sum=(sum-1ll*f[j]*g[j])%mo,su2=(su2+g[j])%mo;
            else su2=(su2-f[j])%mo;
            j--;
        }
        while((k>=l)&&(g[k]>mi)){
            if(j<k)su2=(su2-g[k])%mo;
            else sum=(sum-1ll*f[k]*g[k])%mo,su2=(su2+f[k])%mo;
            k--;
        }
        ans=(ans+sum+1ll*su2*((j<k)?mx:mi)%mo+
            1ll*mx*mi%mo*(md-max(j,k)))%mo;
    }
}
int main(){
    freopen("seq.in","r",stdin);
    freopen("seq.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n)scanf("%d",&a[i]);
    solve(1,n);
    printf("%d",(ans+mo)%mo);
    return 0;
}