bzoj5291: [Bjoi2018]链上二次求和【线段树】

Description

有一条长度为n的链（1≤i < < <script type="math/tex" id="MathJax-Element-8"><</script>n，点i与点i+1之间有一条边的无向图），每个点有一个整数权值，第i个点的权值是a_i。现在有m个操作，每个操作如下：
操作1（修改）：给定链上两个节点u、v和一个整数d，表示将链上u到v唯一的简单路径上每个点权值都加上d。
操作2（询问）：给定两个正整数L、r，表示求链上所有节点个数大于等于L且小于等于r的简单路径节点权值和之和。
由于答案很大，只用输出对质数1000000007取模的结果即可。
一条节点个数为k的简单路径节点权值和为这条上所有k个节点（包括端点）的权值之和，而本题中要求是对所有满足要求的简单路径，求这一权值和的和。
由于是无向图，路径也是无向的，即点1到点2的路径与点2到点1的路径是同一条，不要重复计算。

Input

输入第一行包含两个正整数n、m，分别表示节点个数和操作次数。
第二行包含n个整数，其中第ii个数ai为第ii个点的初始权值。
接下来m行，每行为1 u v d或2 l r的形式，分别表示进行一次操作1（修改）或操作2（询问）。
记操作 1（修改）的次数为 m
n <= 200000,
m <= 500000,
m’<= 100000,
0 <= a_i <1000000007
1 <= u <= n
1<= v <= n
0 <= d < 1000000007
l <= r <= n

Output

对于每次询问，输出一行一个整数，表示答案对1000000007取模的余数。

Sample Input

5 5

1 1 1 1 1

2 5 5

2 1 2

1 1 2 2

2 1 1

1 1 5 3

Sample Output

5

13

9

解题思路：

考虑单次询问的答案为多少($S$为前缀和，$SS$为$S$的前缀和)：

$$ans=\sum\limits_{k=l}^r\sum\limits_{i=k}^{n}(S_i-S_{i-k})$$

$$=\sum\limits_{k=l}^r(\sum\limits_{i=k}^nS_i-\sum\limits_{i=0}^{n-k}S_i)$$ $$=\sum\limits_{k=l}^r(SS_n-SS_{k-1}-SS_{n-k})$$ $$=(r-l+1)SS_n-\sum\limits_{i=l-1}^{r-1}SS_i-\sum\limits_{i=n-r}^{n-l}SS_i$$

所以我们应该用线段树维护$SS$，那么考虑一次修改的影响。
对于 $l\le i\le r，SS_i+=\frac{(i-l+1)(i-l+2)}{2}v$
对于 $i>r，SS_i+=\frac{len(len+1)}{2}v+len(i-r)v，len=r-l+1$
即影响是一个关于 $i$ 的二次函数 ，可以化简为 $ai^2+bi+c$ 的形式，所以我们线段树维护区间$SS$的和及标记 $a,b,c$即可。

我写的代码还是比较短的了……

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=2e5+5,mod=1e9+7,inv2=5e8+4;
inline ll add(ll x,ll y){return x+y>=mod?x+y-mod:x+y;}
inline ll dec(ll x,ll y){return x-y<0?x-y+mod:x-y;}
int n,m;ll ss[N],f1[N],f2[N];
ll sum[N<<2],a[N<<2],b[N<<2],c[N<<2];
void add(int k,int l,int r,ll a1,ll b1,ll c1)
{
    a[k]=add(a[k],a1),b[k]=add(b[k],b1),c[k]=add(c[k],c1);
    sum[k]=add(sum[k],(a1*dec(f2[r],f2[l-1])+b1*dec(f1[r],f1[l-1])+c1*(r-l+1))%mod);
}
void pushdown(int k,int l,int mid,int r)
{
    if(!a[k]&&!b[k]&&!c[k])return;
    add(k<<1,l,mid,a[k],b[k],c[k]);
    add(k<<1|1,mid+1,r,a[k],b[k],c[k]);
    a[k]=b[k]=c[k]=0;
}
void build(int k,int l,int r)
{
    if(l==r){sum[k]=ss[l];return;}
    int mid=l+r>>1;
    build(k<<1,l,mid),build(k<<1|1,mid+1,r);
    sum[k]=add(sum[k<<1],sum[k<<1|1]);
}
void modify(int k,int l,int r,int x,int y,ll a1,ll b1,ll c1)
{
    if(x<=l&&r<=y){add(k,l,r,a1,b1,c1);return;}
    int mid=l+r>>1;pushdown(k,l,mid,r);
    if(x<=mid)modify(k<<1,l,mid,x,y,a1,b1,c1);
    if(y>mid)modify(k<<1|1,mid+1,r,x,y,a1,b1,c1);
    sum[k]=add(sum[k<<1],sum[k<<1|1]);
}
ll query(int k,int l,int r,int x,int y)
{
    if(x==l&&y==r)return sum[k];
    int mid=l+r>>1;pushdown(k,l,mid,r);
    if(y<=mid)return query(k<<1,l,mid,x,y);
    else if(x>mid)return query(k<<1|1,mid+1,r,x,y);
    else return add(query(k<<1,l,mid,x,mid),query(k<<1|1,mid+1,r,mid+1,y));
}
void Modify(int l,int r,ll v)
{
    if(l>r)swap(l,r);
    ll a1=v,b1=dec(3,add(l,l))*v%mod,c1=((ll)l*l-3*l+2)%mod*v%mod;
    modify(1,0,n,l,r,a1,b1,c1);
    if(r==n)return;ll len=r-l+1;
    a1=0,b1=2*len*v%mod,c1=add(len*(len+1-2*r)%mod,mod)*v%mod;
    modify(1,0,n,r+1,n,a1,b1,c1);
}
ll Query(int l,int r)
{
    ll res=dec(dec(query(1,0,n,n,n)*(r-l+1)%mod,query(1,0,n,l-1,r-1)),query(1,0,n,n-r,n-l));
    return res*inv2%mod;
}
int main()
{
    //freopen("sum.in","r",stdin);
    //freopen("sum.out","w",stdout);
    n=getint(),m=getint();
    for(int i=1;i<=n;i++)ss[i]=add(getint(),ss[i-1]);
    for(int i=1;i<=n;i++)ss[i]=add(ss[i],ss[i-1]);
    for(int i=1;i<=n;i++)ss[i]=add(ss[i],ss[i]),f1[i]=add(f1[i-1],i),f2[i]=add(f2[i-1],(ll)i*i%mod);
    build(1,0,n);int op,l,r;
    while(m--)
    {
        op=getint(),l=getint(),r=getint();
        if(op==1)Modify(l,r,getint());
        else printf("%lld\n",Query(l,r));
    }
    return 0;
}