[JZOJ6247]【NOI2019模拟2019.6.27】C【计数】-CSDN博客

Description

在这里插入图片描述
n<=200000

Solution

比赛时没做出这道题真的太弟弟了
首先我们从小到大插入数i，考虑B中有多少个区间的最大值为i
恰好出现的次数不太好计算，我们考虑计算最大值小于等于i，再做一个差分即可。

然后直接分成长度在一段内的和长度跨过一段边界的考虑，跨过完整的一段的区间的答案一定是整个序列最大值

分类讨论即可，式子并不难推，有一个地方可以直接暴力计算前缀和。

复杂度O(N)或加上O(MAX(a))

Code

#include <bits/stdc++.h>
#define fo(i,a,b) for(int i=a;i<=b;++i)
#define fod(i,a,b) for(int i=a;i>=b;--i)
typedef long long LL;
const int mo=1000000007;
const int N=200005;
using namespace std;
int n,a[N],ft[N],sz[N],d[N];
LL ans[N];
int getf(int k)
{
    return (!ft[k]||ft[k]==k)?k:ft[k]=getf(ft[k]);
}
void merge(int x,int y)
{
    x=getf(x),y=getf(y);
    ft[y]=x,sz[x]+=sz[y];
}
bool cmp(int x,int y)
{
    return (a[x]<a[y])||(a[x]==a[y]&&x<y);
}
LL sm[N],sp[N];
LL calc(int x,int y)
{
    LL s=0,ml=min(x+1,y);
    s=(s+sp[x+y-1]-((x+y-2*ml-1<0)?0:sp[x+y-2*ml-1])+mo)%mo;
    s=(s-sm[x]+((x<ml)?0:sm[x-ml])+mo)%mo;
    s=(s-sm[y-1]+((y-1<ml)?0:sm[y-1-ml])+mo)%mo;
    return s;
}
int main()
{
    cin>>n;
    fo(i,1,n) scanf("%d",&a[i]),ft[i]=i,d[i]=i;
    sort(d+1,d+n+1,cmp);
    ans[0]=0;
    fo(i,1,n) sm[i]=(sm[i-1]+(LL)i*(i+1)/2)%mo;
    sp[1]=1;
    fo(i,2,n) sp[i]=(sp[i-2]+(LL)i*(i+1)/2)%mo;
    calc(1,4);
    fo(i,1,n-1)
    {
        int w=d[i];
        ans[i]=ans[i-1];
        if(sz[n]&&sz[1]) ans[i]=(ans[i]-calc(sz[getf(n)],sz[getf(1)])+mo)%mo; 
        sz[w]=1;
        if(sz[w-1])
        {
            ans[i]-=sm[sz[getf(w-1)]];
            merge(w-1,w);
        }
        if(sz[w+1])
        {
            ans[i]-=sm[sz[getf(w+1)]];
            merge(w,w+1);
        }
        ans[i]=(ans[i]%mo+sm[sz[getf(w)]])%mo;
        if(sz[n]&&sz[1]) ans[i]=(ans[i]+calc(sz[getf(n)],sz[getf(1)]))%mo; 
    }   
    LL c=(LL)n*(n+1)/2%mo;
    ans[n]=c*(c+1)%mo*((mo+1)/2)%mo;
    LL s1=0;
    fo(i,1,n) 
    {
        s1=(s1+(LL)(ans[i]-ans[i-1]+mo)%mo*a[d[i]])%mo;
    }
    printf("%lld\n",s1);
}

转载于:https://www.cnblogs.com/BAJimH/p/11123500.html