【模板】一维&二维 前缀和 差分

前言

前缀和是一种重要的预处理，能大大降低查询的时间复杂度，而差分则是一种和前缀和相对的策略。

一维前缀和

例题：给定一个长度为n的数组A，求区间[L，R]之内的数组元素的和。
分析：
1.暴力解法必然是二重循环，时间复杂度较高；
2.考虑维护一个数组sum，sum[i]表示数组A的前i项的求和，那么sum[R]-sum[L-1]即为所求答案。

int n,a[100000+10],sum[100000+10];
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+a[i];  //前缀和
}

一维前缀和例题：洛谷p1115-最大字段和

一维差分

例题：给定一个长度为n的数组A，先进行p次操作，每次操作对【L，R】内的数组元素加k，之后询问q次，每次询问【L，R】内数组元素的和。
分析：用一个数组dif来储存数组元素值的变化，di在这里插入代码片f[i]=k表示记录下数组A【i，n】内的值加k。

int n,p,a[10000+10],sum[10000+10],dif[10000+10];
int main()
{
    cin>>n>>p;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    for(int i=1;i<=p;i++)
    {
        int L,R,k;
        cin>>L>>R>>k;
        dif[L]+=k;     //L往后的值都要加上k
        dif[R+1]-=k;   //但是R往后的值不能加k，所以要在上述基础上减去k
    }
    int add = 0; //表示从前往后累计加的值
    for(int i=1;i<=n;i++)
    {  //为了便于理解，我们首先把原数组的值求出来
        add+=dif[i];
        a[i]+=add;
    }
    for(int i=1;i<=n;i++)
        sum[i]=sum[i-1]+a[i]; //然后再去求前缀和
}

二维前缀和

多维前缀和基于容斥原理进行计算
给定一个二维数组，a[i][j]，求从[1][1]到[i][j]的所有数组元素的求和。

int n,m,a[1000][1000],sum[1000][1000];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        cin>>a[i][j];
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        sum[i][j] = sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j];
}

二维差分

在二维前缀和例题基础上，给定p个操作，每次操作对[x1][y1]到[x2][y2]这个子矩阵的所有元素值加1，然后q次询问，询问[x1][y1]到[x2][y2]这个子矩阵的所有元素的和。

int main()
{
    cin>>n>>m>>k>>q;
    for(int i=1;i<=k;i++)
    {
        int x1,x2,y1,y2;
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);
        dif[x1][y1]++;       //差分计算，画图易得，表示从[i][j]到[n][m]子矩阵的所有元素加1
        dif[x1][y2+1]--;
        dif[x2+1][y1]--;
        dif[x2+1][y2+1]++;
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)//根据差分，求矩阵的每一个点的元素值（仍然是前缀和形式）
        a[i][j] = a[i-1][j]+a[i][j-1]-a[i-1][j-1]+dif[i][j];//即差分的前缀和是矩阵元素
        
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++) //对每个数组元素求二维前缀和
        sum[i][j] = sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1]+a[i][j]; 

    for(int i=1;i<=q;i++)
    {
        int x1,x2,y1,y2;
        scanf("%d%d%d%d",&x1,&y1,&x2,&y2);    //定区域查询
        ll ans = sum[x2][y2]-sum[x2][y1-1]-sum[x1-1][y2]+sum[x1-1][y1-1];
        printf("%lld\n",ans);
    }

参考资料：
前缀和 & 差分
算法2-1 前缀和与差分