LibreOJ #2512.「BJOI2018」链上二次求和 线段树

题意

有一个长度为n的序列a和m个操作，每个操作形如
操作1（修改）：给定链上两个节点u、v和一个整数d，表示将链上u到v唯一的简单路径上每个点权值都加上d。
操作2（询问）：给定两个正整数L、r，表示求链上所有节点个数大于等于L且小于等于r的简单路径节点权值和之和。
n≤200000,m≤500000 n ≤ 200000 , m ≤ 500000

分析

我们设 Si S i 表示前缀和， SSi S S i 表示前缀和的前缀和，那么要求的就是

∑i=lr∑j=lnSj−Sj−i ∑ i = l r ∑ j = l n S j − S j − i

=∑i=lr(∑j=lnSj−∑j=0n−iSj) = ∑ i = l r ( ∑ j = l n S j − ∑ j = 0 n − i S j )

=∑i=lr(SSn−SSl−1−SSn−i) = ∑ i = l r ( S S n − S S l − 1 − S S n − i )

=(r−l+1)SSn−∑i=l−1r−1SSi−∑i=n−rn−lSSi = ( r − l + 1 ) S S n − ∑ i = l − 1 r − 1 S S i − ∑ i = n − r n − l S S i

那现在问题就在于如何维护 SS S S 。
考虑一次修改 (l,r,v) ( l , r , v ) 对 SS S S 的贡献。
对所有的 l≤i≤r l ≤ i ≤ r ，有

SSi+=(i−l+1)(i−l+2)2v S S i + = ( i − l + 1 ) ( i − l + 2 ) 2 v

对所有的 r<i r < i ，有

SSi+=len(len+1)2v+len(i−r)v S S i + = l e n ( l e n + 1 ) 2 v + l e n ( i − r ) v

其中 len=r−l+1 l e n = r − l + 1 。
不难发现对 SSi S S i 的贡献都是关于 i i 的一个二次函数，直接用线段树维护即可。

代码

#include
#include
#include
#include
#include
#define ls d*2
#define rs d*2+1

typedef long long LL;

const int N=200005;
const int MOD=1000000007;
int ny2=(MOD+1)/2;

int n,m,a[N];
struct tree{int l,r,s,s0,s1,s2,tag0,tag1,tag2;}t[N*4];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void build(int d,int l,int r)
{
    if (l==r) {t[d].s=a[l];t[d].s1=l;t[d].s2=(LL)l*l%MOD;t[d].s0=1;return;}
    int mid=(l+r)/2;
    build(ls,l,mid);build(rs,mid+1,r);
    t[d].s=(t[ls].s+t[rs].s)%MOD;
    t[d].s0=(t[ls].s0+t[rs].s0)%MOD;
    t[d].s1=(t[ls].s1+t[rs].s1)%MOD;
    t[d].s2=(t[ls].s2+t[rs].s2)%MOD;
}

void mark(int d,int a,int b,int c)
{
    (t[d].tag0+=a)%=MOD;
    (t[d].tag1+=b)%=MOD;
    (t[d].tag2+=c)%=MOD;
    (t[d].s+=(LL)a*t[d].s0%MOD)%=MOD;
    (t[d].s+=(LL)b*t[d].s1%MOD)%=MOD;
    (t[d].s+=(LL)c*t[d].s2%MOD)%=MOD;
}

void pushdown(int d)
{
    int a=t[d].tag0,b=t[d].tag1,c=t[d].tag2;
    if (!a&&!b&&!c) return;
    t[d].tag0=t[d].tag1=t[d].tag2=0;
    mark(ls,a,b,c);
    mark(rs,a,b,c);
}

void ins(int d,int l,int r,int x,int y,int a,int b,int c)
{
    if (yx>r) return;
    if (lif (x<=l&&r<=y) {mark(d,a,b,c);return;}
    int mid=(l+r)/2;
    if (x<=mid) ins(ls,l,mid,x,y,a,b,c);
    if (y>mid) ins(rs,mid+1,r,x,y,a,b,c);
    t[d].s=(t[ls].s+t[rs].s)%MOD;
}

int query(int d,int l,int r,int x,int y)
{
    if (yx>r) return 0;
    if (lif (x<=l&&r<=y) return t[d].s;
    int mid=(l+r)/2;
    if (y<=mid) return query(ls,l,mid,x,y);
    else if (x>mid) return query(rs,mid+1,r,x,y);
    else return (query(ls,l,mid,x,y)+query(rs,mid+1,r,x,y))%MOD;
}

int main()
{
    n=read();m=read();
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++) (a[i]+=a[i-1])%=MOD;
    for (int i=1;i<=n;i++) (a[i]+=a[i-1])%=MOD;
    build(1,1,n);
    while (m--)
    {
        int op=read(),l=read(),r=read();
        if (l>r) std::swap(l,r);
        if (op==1)
        {
            int v=read(),len=r-l+1;
            ins(1,1,n,l,r,(LL)(l-1)*(l-2)%MOD*ny2%MOD*v%MOD,-(LL)ny2*(l*2-3)%MOD*v%MOD,(LL)ny2*v%MOD);
            ins(1,1,n,r+1,n,(LL)(len+1)*len%MOD*ny2%MOD*v%MOD-(LL)r*v%MOD*len%MOD,(LL)v*len%MOD,0);
        }
        else
        {
            int ans=(LL)(r-l+1)*query(1,1,n,n,n)%MOD;
            (ans-=query(1,1,n,l-1,r-1))%=MOD;
            (ans-=query(1,1,n,n-r,n-l)%MOD)%=MOD;
            printf("%d\n",(ans+MOD)%MOD);
        }
    }
    return 0;
}