【C/C++】前缀和和差分

简单来说

前缀和是指某序列的前n项和，可以把它理解为数学上的数列的前n项和，而差分可以看成前缀和的逆运算。合理的使用前缀和与差分，可以将某些复杂的问题简单化。

而差分是前缀和的逆运算
b序列是a序列的前缀和 则a序列就是b序列的差分

一维前缀和

一维前缀和顾名思义就是一维数组中的前缀和
代码实现

#include<iostream>
using namespace std;
const int N = 100010;
int q[N],s[N];
//q数组是原数组
//s数组是前缀和数组
int main()
{
    int n;
    scanf("%d%d",&n);
    for(int i = 1; i <= n;i ++)//因为存在对数组下标-1调用的情况 
    //我们将元素从第一个位置开始存放
    {
        scanf("%d",&q[i]);
    }
    for(int i = 1; i <= n ;i++)
    {
        s[i] = s[i - 1] + q[i];
        //每个前缀和都是上个前缀和加上这个位置原数组的元素
    }
    return 0;
}

例题

求一段区间的区间和

正常我们的做法就是遍历数组 时间复杂度为o(n)

但是使用前缀和我们就只需要用r位置的前缀和减去l位置的前缀和就可以得到这段区间的和 时间复杂度只需要o(1)

#include<iostream>
using namespace std;
const int N = 100010;
int q[N],s[N];

int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i = 1; i <= n;i ++)
    {
        scanf("%d",&q[i]);
    }
    for(int i = 1; i <= n ;i++)
    {
        s[i] = s[i - 1] + q[i];
    }
    while(m--)
    {
        int i ,j;
        scanf("%d%d",&i,&j);
        printf("%d\n",s[j]-s[i-1]);
    }
    return 0;
}

二维前缀和

二位前缀和顾名思义就是在二维数组里的前缀和

例如
存在一个二维数组a[10][10] 我们想要求a[3][3]的前缀和 应该怎么求呢？

由图可知 应该是该点左上部分的面积

我们可以用数学知识来解决这个问题

1. 求出该点左区域的面积
2. 求出该点上区域的面积
3. 加上该点的原数组的点的数值
4. 减去左上部分多加的元素

最后可以得出公式
如果要求该点的前缀和 那么就可以用公式

例题

这道题需要我们求出一块矩形区域的和

例如这块红色矩形的和
我们定义左上角为（x1,y1）右下角为（x2,y2）

我们可以使用上面类似的方法

1. 求出b[x2][y2]
2. 减去b[x2][y1-1]
3. 减去b[x1-1][y1];
4. 加上多减去的部分b[x1-1][y1-1];

#include <iostream>

using namespace std;

const int N = 1010;

int n, m, q;
int s[N][N];

int main()
{
    scanf("%d%d%d", &n, &m, &q);

    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ )
            scanf("%d", &s[i][j]);//输入原数组

    for (int i = 1; i <= n; i ++ )
        for (int j = 1; j <= m; j ++ )
            s[i][j] += s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1];//用刚刚我们列出的公式求前缀和二维数组

    while (q -- )
    {
        int x1, y1, x2, y2;
        scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
        printf("%d\n", s[x2][y2] - s[x1 - 1][y2] - s[x2][y1 - 1] + s[x1 - 1][y1 - 1]);
        //用公式求一块矩形的面积
    }

    return 0;
}

一维差分

#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int a[N], b[N];
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        b[i] = a[i] - a[i - 1];      //构建差分数组
    }
    return 0;
}

例题

**因为我们构建的b数组的前缀和就是a数组 所以我们在b数组l位置加一个c
根据公式可得l位置之后的每个a都会+c **

此时我们在向r+1的位置让b减去c，这样从r+1位置开始后面的所有a都减去c 这样就达到了在l到r区间所有a加上c的操作

这种做法时间复杂度为o(1) 而传统遍历做法时间复杂度为o(n)

#include<iostream>
using namespace std;
const int N = 1e5 + 10;
int a[N], b[N];
int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        b[i] = a[i] - a[i - 1];      //构建差分数组
    }
    int l, r, c;
    while (m--)
    {
        scanf("%d%d%d", &l, &r, &c);
        b[l] += c;     //将序列中[l, r]之间的每个数都加上c
        b[r + 1] -= c;
    }
    for (int i = 1; i <= n; i++)
    {
        a[i] = b[i] + a[i - 1];    //前缀和运算
        printf("%d ", a[i]);
    }
    return 0;
}

二维差分

代码

#include<iostream>
using namespace std;

const int N = 1010;
int a[N][N],b[N][N];
int m , n, q;

void insert(int x1, int y1, int x2,int y2, int c)//插入函数
{
    b[x1][y1] += c;
    b[x2 + 1][y1] -= c;
    b[x1][y2 + 1] -= c;
    b[x2 + 1][y2 + 1] += c;
}
int main()
{
    scanf("%d%d%d",&n,&m,&q);
    for(int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            scanf("%d", &a[i][j]);输入原数组的值
            insert(i, j, i, j, a[i][j]);//构造差分数组b
        }
    }
    
    while(q --)
    {
        int x1, y1, x2, y2, c;
        scanf("%d%d%d%d%d",&x1, &y1, &x2, &y2, &c);
        insert(x1, y1, x2, y2, c);
    }
    
    for(int i = 1; i <= n; i++)
    {
        for (int j = 1; j <= m; j++)
        {
            b[i][j] = b[i][j - 1] + b[i - 1][j] + b[i][j] - b[i - 1][j - 1];
            printf("%d ",b[i][j]);
        }
        printf("\n");
    }
    return 0;
}