差分与树状数组

一维树状数组

长度为$n$的序列: $a_1,a_2,\cdots ,a_n$

一阶差分与前缀和

一阶差分数组: $d_i=a_i-a_{i-1}(a_0=0)$

考虑序列$a$第$m$项:

$a_m=\sum _{i=1}^m{d}_i$

考虑序列$a$前$m$项和:

$\sum _{i=1}^m{a}_i=\sum _{i=1}^m\sum _{j=1}^i{d}_j=\sum _{i=1}^m(m+1-i)\cdot d_i=(m+1)\sum _{i=1}^m{d}_i-\sum _{i=1}^{m}i\cdot d_i$

//区间加常数
for(int i=l;i<=r;++i)
    a[i]+=v;
//查询区间和
for(int i=l;i<=r;++i)
    sum+=a[i];
return sum;

树状数组优化上面的代码, 复杂度: $O(n\log n)$

struct BinaryIndexTree
{
    static const int __=1e5+5;
    typedef ll type;
    int n;type a[2][__];

    void add(int x,type v)
    {
        for(int i=x;i<=n;i+=i&-i)
        {
            a[0][i]+=v;
            a[1][i]+=x*v;
        }
    }

    void add(int l,int r,type v)
    {
        if(l>r)return;
        add(l,v),add(r+1,-v);
    }

    type sum(int x)
    {
        type b[2]={0};
        for(int i=x;i;i-=i&-i)
        {
            b[0]+=a[0][i];
            b[1]+=a[1][i];
        }
        return (x+1)*b[0]-b[1];
    }

    type sum(int l,int r)
    {
        if(l>r)return 0;
        return sum(r)-sum(l-1);
    }
}B;

二阶差分与前缀和

二阶差分数组: $e_i=d_i-d_{i-1}=a_i-2\cdot a_{i-1}+a_{i-2}(a_{-1}=a_0=0)$

考虑序列$a$第$m$项:

$a_m=\sum _{i=1}^m{d}_i=\sum _{i=1}^m\sum _{j=1}^i{e}_j=(m+1)\sum _{i=1}^m{e}_i-\sum _{i=1}^{m}i\cdot e_i$

考虑序列$a$前$m$项和:

$\begin{array}{rl}\sum _{i=1}^m{a}_i& =\sum _{i=1}^m((i+1)\sum _{j=1}^i{e}_j-\sum _{j=1}^{i}j\cdot e_j)\\ & =\left(\sum _{i=1}^m{e}_i\sum _{j=i+1}^{m+1}j\right)-(\sum _{i=1}^m{e}_i\cdot ((m+1-i)\cdot i\left)\right)\\ & =(\sum _{i=1}^m{e}_i\cdot \frac{(m+2+i)\cdot (m+1-i)}{2})-(\sum _{i=1}^m{e}_i\cdot ((m+1-i)\cdot i\left)\right)\\ & =\sum _{i=1}^m{e}_i\cdot \frac{(m+2-i)\cdot (m+1-i)}{2}\\ & =\sum _{i=1}^m{e}_i\cdot \frac{(m^2+3m+2)-(2m+3)\cdot i+i^2}{2}\\ & =\frac{(m+1)(m+2)}{2}\sum _{i=1}^m{e}_i-\frac{2m+3}{2}\sum _{i=1}^{m}i\cdot e_i+\frac{1}{2}\sum _{i=1}^m{i}^2\cdot e_i\end{array}$

//区间加一次函数
for(int i=l;i<=r;++i)
    a[i]+=(i-l)*k+b;
//查询区间和
for(int i=l;i<=r;++i)
    sum+=a[i];
return sum;

树状数组优化以上代码, 复杂度: $O(n\log n)$

struct BinaryIndexTree
{
    static const int __=1e5+5;
    typedef ll type;
    int n;type a[3][__];

    void add(int x,type v)
    {
        for(int i=x;i<=n;i+=i&-i)
        {
            a[0][i]+=v;
            a[1][i]+=x*v;
            a[2][i]+=x*x*v;
        }
    }
    //a[i]+=b+k*(i-l)
    void add(int l,int r,type k,type b)
    {
        if(l>r)return;
        add(l,b);
        add(l+1,k-b);
        add(r+1,-b-(r-l+1)*k);
        add(r+2,b+(r-l)*k);
    }

    type sum(int x)
    {
        type b[3]={0};
        for(int i=x;i;i-=i&-i)
        {
            b[0]+=a[0][i];
            b[1]+=a[1][i];
            b[2]+=a[2][i];
        }
        return ((x+1)*(x+2)*b[0]-(2*x+3)*b[1]+b[2])/2;
    }

    type sum(int l,int r)
    {
        if(l>r)return 0;
        return sum(r)-sum(l-1);
    }
}B;

$k$阶差分与前缀和

不难看出: 在维护原数组$a_i$的$k$阶差分数组$b_i$时, $a_i$的前$m$项和都是如下几个相同形式的部分求和

$\sum f(m)\cdot \sum _{i=1}^m{i}^c\cdot b_i(f(x)$是关于$x$的多项式$)$

考虑一个部分的前$m$项和:

$\begin{array}{rl}\sum _{i=1}^{m}f(i)\sum _{j=1}^i{j}^c\cdot b_j& =\sum _{i=1}^m{i}^c\cdot b_i\sum _{j=i}^{m}f(j)=\sum _{i=1}^m{i}^c\cdot b_i\cdot (\sum _{j=1}^{m}f(j)-\sum _{j=1}^{i-1}f(j))\\ & =\sum _{i=1}^m{i}^c\cdot b_i\cdot (g(m)-g(i-1))(g(n)=\sum _{i=1}^{n}f(i))\end{array}$

运用上述方法不难求得三阶差分数组$f_i$与前缀和的关系

$\sum _{i=1}^m{a}_i=\frac{(m+1)(m+2)(m+3)}{6}\sum _{i=1}^m{f}_i-\frac{3m^2+12m+11}{6}\sum _{i=1}^{m}i\cdot f_i+\frac{m+2}{2}\sum _{i=1}^m{i}^2\cdot f_i-\frac{1}{6}\sum _{i=1}^m{i}^3\cdot f_i$

//区间加二次函数
for(int i=l;i<=r;++i)
    a[i]+=(i-l)*(i-l)*x+(i-l)*y+z;
//查询区间和
for(int i=l;i<=r;++i)
    sum+=a[i];
return sum;

树状数组优化以上代码, 复杂度: $O(n\log n)$

struct BinaryIndexTree
{
    static const int __=1e5+5;
    typedef ll type;
    int n;type a[4][__];

    void add(int x,type v)
    {
        for(int i=x;i<=n;i+=i&-i)
        {
            a[0][i]+=v;
            a[1][i]+=x*v;
            a[2][i]+=x*x*v;
            a[3][i]+=x*x*x*v;
        }
    }
    //a[i]+=x*(i-l)*(i-l)+y*(i-l)+z;
    void add(int l,int r,type x,type y,type z)
    {
        if(l>r)return;
        add(l,z);
        add(l+1,x+y-2*z);
        add(l+2,x-y+z);
        type p=r-l+1,q=r-l;
        add(r+1,-p*p*x-p*y-z);
        add(r+2,(p*p+q*q-2)*x+(p+q)*y+2*z);
        add(r+3,-q*q*x-q*y-z);
    }

    type sum(int x)
    {
        type b[4]={0};
        for(int i=x;i;i-=i&-i)
        {
            b[0]+=a[0][i];
            b[1]+=a[1][i];
            b[2]+=a[2][i];
            b[3]+=a[3][i];
        }
        return ((x+1)*(x+2)*(x+3)*b[0]-(3*x*x+12*x+11)*b[1]+(3*x+6)*b[2]-b[3])/6;
    }

    type sum(int l,int r)
    {
        if(l>r)return 0;
        return sum(r)-sum(l-1);
    }
}B;