前缀和 差分

前缀和 差分

差分与前缀和为互逆运算，适用于区间修改+区间查询问题，可将区间修改转换为单点修改：$\underset{差分}{d}\stackrel{\sum }{⟶}\underset{原}{a}\stackrel{\sum }{⟶}\underset{前缀和}{s}$

实现时通常数组下标从1开始，避免越界访问到负下标

一维前缀和 一维差分

几何意义：一维为线性的前缀运算，前缀和为长度

• 一维前缀和:$s_i=s_{i-1}+a_i\Rightarrow a_i=s_i-s_{i-1}$
• 第一类一维差分:$d_i=a_i-a_{i-1}\Rightarrow a_i=a_{i-1}+d_i$

int n,a[n+1],s[n+1],d[n+2];
void init(){
    for(int i=1;i<=n;i++){
        s[i]=s[i-1]+a[i];
        d[i]=a[i]-a[i-1];
    }
}

预处理的复杂度为$O(n)$

区间修改

设操作区间为$[L,R]$：

• 差分数组中$d_L+=\Delta$的单点修改，等价于原数组$a_i+=\Delta (i\in [L,+\infty ])$的区间修改
• 若操作区间存在右端点$R$，则需在$R+1$处撤销修改:$d_{R+1}-=\Delta$，相当于原数组$a_i-=\Delta (i\in [R+1,+\infty ])$

void modify(int l,int r,int d){
    d[l]+=d,d[r+1]-=d;
}

区间修改的复杂度为$O(1)$

区间查询

• 查询原数组：第二类一维差分：$d$初始化全部为0，用于存储原数组的每个元素的增量$\Delta$。区间修改$[L,R]$方法不变，之后求前缀和即为原数组每个元素的增量$\Delta$

• 查询前缀和：$\sum _{i=L}^R{a}_i=\sum _{i=1}^R{a}_i-\sum _{i=1}^{L-1}{a}_i$。

int query(int l,int r){
    //第一类一维差分
    for(int i=1;i<=n;i++){
        a[i]=d[i]+a[i-1];//更新原数组
        s[i]=a[i]+s[i-1];
	}
	return s[r]-s[l-1];
    //第二类一维差分
    for(int i=1;i<=n;i++){
        d[i]+=d[i-1];
    }
    for(int i=1;i<=n;i++){
        a[i]+=d[i];
        s[i]=s[i-1]+a[i];
    }
    return s[r]-s[l-1];
}

区间查询复杂度为$O(n)$。由此可见差分数组擅长处理区间修改，但不擅长单点查询。

二维前缀和 二维差分

几何意义：二维为面的前缀运算，前缀和为面积

• 二维前缀和：$s[i][j]=\underset{上侧所有部分}{s[i-1][j]}+\underset{左侧所有部分}{s[i][j-1]}-\underset{左上侧被加2次部分}{s[i-1][j-1]}+\underset{本身}{a[i][j]}$

• 矩阵前缀和：设矩阵左上顶点为$(x,y)$，右下顶点为$(i,j)$：$S=s[i][j]-\underset{上侧非矩形部分}{s[x-1][j]}-\underset{左方非矩形部分}{s[i][y-1]}+\underset{左上方被减2次非矩形部分}{s[x-1][y-1]}$

• 第一类二维差分：$d[i][j]=a[i][j]-\underset{上侧所有部分}{a[i-1][j]}-\underset{左侧所有部分}{a[i][j-1]}+\underset{左上方被减2次部分}{a[i-1][j-1]}$

二维差分较为抽象，从前缀和与差分为互逆运算角度理解

int n,m,a[n+1][m+1],s[n+1][m+1],d[n+1][m+1];//n行m列
void init(){
    for(int i=1;j<=n;j++)
        for(int j=1;i<=m;i++){
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
            d[i][j]=a[i][j]-a[i-1][j]-a[i][j-1]+a[i-1][j-1];//第一类二维差分
        }
}

区间修改

设操作矩阵左上角坐标为$[x_1,y_1]$，右下角坐标为$[x_2,y_2]$:

差分数组对$d[i][j]+=\Delta$的单点修改，等价于在原数组中的$[i,j]$到$[+\infty ,+\infty ]$范围的区间修改。因此，$[x_1,y_1]$到$[x_2,y_2]$内中每个元素加$\Delta$:

1. $d[y_1][x_1]+=\Delta$：原数组$[y_1,x_1]$到$[+\infty ,+\infty ]+\Delta$的二维区间修改
2. $d[y_2+1][x_1]-=\Delta$：原数组$[y_2+1,x_1]$到$[+\infty ,+\infty ]$矩形中元素$-\Delta$(矩形右侧无需修改)
3. $d[y_1][x_2+1]-=\Delta$：原数组$[y_1,x_2+1]$到$[+\infty ,+\infty ]$矩形中元素$-\Delta$(矩形下方无需修改)
4. $d[y_2+1][x_2+1]+=\Delta$：原数组$[y_2+1,x_2+1]$到$[+\infty ,+\infty ]$矩形中元素被减了2次$\Delta$，撤销一次修改(矩形右下方)

void modify(int x1,int y1,int x2,int y2,int d){
    d[y1][x1]+=d;
    d[y2+1][x1]-=d;
    d[y1][x2+1]-=d;
    d[y2+1][x2+1]+=d;
}

区间查询

• 查询原数组：第二类二维差分：$d$全部初始化为0，用于存储原数组内每个元素的增量$\Delta$。区间修改方法不变，之后求前缀和即为原数组每个元素的增量$\Delta$。
• 查询前缀和

int query(int x1,int y1,int x2,int y2){
    //第一类差分数组
    for(int j=1;j<=n;j++)
        for(int i=1;i<=m;i++){
            a[i][j]=a[i-1][j]+a[i][j-1]-a[i-1][j-1]+d[i][j];//重新计算原数组
            s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
        }
    return s[y2][x2]-s[y1-1][x2]-s[y2][x1-1]+s[y1][x1];
    //第二类差分数组
    //第一类差分数组
    for(int j=1;j<=n;j++)
        for(int i=1;i<=m;i++)
            d[i][j]+=d[i-1][j]+d[i][j-1]-d[i-1][j-1];
    for(int j=1;j<=n;j++)
        for(int i=1;i<=m;i++)
            a[i][j]+=d[i][j];
}

• 技巧：等价于将原数组所有值视为$d$数组增量。元素初始全为0，进行$\sum _{i=1}^n\sum _{j=1}^m$循环:

  • $d[i][j]+=a[i][j]$

  • $d[i+1][j]-=a[i][j]$

  • $d[i][j+1]-=a[i][j]$

  • $d[i+1][j+1]+=a[i][j]$

for(int j=1;j<=n;j++){
    for(int i=1;i<=m;i++){
        d[j][i]+=a[j][i];
        d[j+1][i]-=a[j][i];
        d[j][i+1]-=a[j][i];
        d[j+1][i+1]+=a[j][i];
    }
}

更高维的前缀和：三维前缀和，为对空间的前缀运算，几何意义为体积