树状数组 区间修改

树状数组 区间修改

利用单点修改我们可以做到
1.单点加, 询问区间和
2.区间加, 询问单点值

下面是关于树状数组的区间修改, 询问区间和
假设: $a_i$是储存原数据的数组
令: $x_i=a_i-a_{i-1}\ \ \ \ \ \ \ \ \ \ \ \ \ \ \ \ (a_0=0)$
那么:
$x_n+x_{n-1}+\cdots+x_1=(a_n-a_{n-1})+(a_{n-1}-a_{n-2})+\cdots+(a_1-a_0)$
右边消去得到: $a_n=x_n+x_{n-1}+\cdots+x_1$
即:

$$a_n=\sum_{i=1}^nx_n$$
那么:
$$\begin{align}a_1+a_2+\cdots+a_n &=(x_1)+(x_1+x_2)+\cdots+(x_1+x_2+\cdots+x_n)\\ &=n(x_1+x_2+\cdots+x_n)-\big(0\cdot x_1+1\cdot x_2+\cdots+(n-1)\cdot x_n\big) \end{align}$$

在区间 $l$ 到 $r$ 所有数$+val$
相当于在$x_l$位置$+val$, 在$x_{r+1}$位置$-val$
又转换成了树状数组单点修改问题

用树状数组分别维护 $x_i$ 与 $(i-1)\cdot x_i$这$2$个序列的前缀和
假设:

$$c_n=x_1+x_2+\cdots+x_n$$

$$d_n=0\cdot x_1+1\cdot x_2+\cdots+(n-1)\cdot x_n$$

区间更新/询问区间和代码:

int n,c[100005],d[100005];

void update(int l,int r,int val)
{
    for(int i=l--; i<=n; i+=i&(-i))
        c[i]+=val,d[i]+=l*val;
    for(int i=r+1; i<=n; i+=i&(-i))
        c[i]-=val,d[i]-=r*val;
}

int get_sum(int l,int r)
{
    int res=0;
    for(int i=r; i; i-=i&(-i))
        res+=r*c[i]-d[i];
    for(int i=--l; i; i-=i&(-i))
        res-=l*c[i]-d[i];
    return res;
}