算法学习系列（八）：前缀和与差分

引言

前缀和与差分其实是一对逆序关系，然后前缀和的话，在算法题中用的还是比较多的，用的目的还是为了降低时间复杂度，前缀和主要是用来快速求 $[l,r]$ 的值，而差分则是快速在 $[l,r]$ 中插入数 $c$ 。

一、前缀和

1.概念

前缀和：能用 $O(1)$ 的速度求出数组 $a$ 中 $[l,r]$ 的和
原理：用一个数组 $s[i]$ ，代表从 $a[1]\sim a[i]$ 的和，所以 $a$ 中 $[l,r]$ 的和为 $s[r]-s[l-1]$
注意的一点是前缀和下标要从 $1$ 开始，这样每个 $s[l,r]=s[r]-s[l-1]$ ，比如说 $[1,10]$ 的话就为 $s[10]-s[0]$ 这样让每个公式都是一样的，不需要特判

2. 一维前缀和例题

输入一个长度为n的整数序列。接下来再输入 m个询问，每个询问输入一对 l,r。
对于每个询问，输出原序列中从第 l个数到第 r个数的和。

输入格式
第一行包含两个整数 n和 m。第二行包含 n个整数，表示整数数列。接下来 m行，每行包含两个整数 l和 r，表示一个询问的区间范围。

输出格式共 m行，每行输出一个询问的结果。数据范围1≤l≤r≤n,1≤n,m≤100000,−1000≤数列中元素的值≤1000
输入样例：
5 3
2 1 3 6 4
1 2
1 3
2 4
输出样例：
3
6
10

#include <cstdio>
#include <iostream>

using namespace std;

const int N = 100010;

int s[N];
int n, m;

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i)
    {
        int t;
        scanf("%d", &t);
        s[i] = s[i-1] + t;  //这里直接将s当作a了，否则就要重新循环算一遍，可以直接在这里算出s[i]来
    }
    
    while(m--)
    {
        int l, r;
        scanf("%d%d", &l, &r);
        printf("%d\n", s[r] - s[l-1]);
    }
    
    return 0;
}

3. 二维前缀和例题

输入一个 n行 m列的整数矩阵，再输入 q个询问，每个询问包含四个整数 x1,y1,x2,y2，表示一个子矩阵的左上角坐标和右下角坐标。对于每个询问输出子矩阵中所有数的和。

输入格式
第一行包含三个整数 n，m，q。接下来 n行，每行包含 m个整数，表示整数矩阵。接下来 q行，每行包含四个整数 x1,y1,x2,y2，表示一组询问。

输出格式
共 q行，每行输出一个询问的结果。

数据范围
1≤n,m≤1000,1≤q≤200000,1≤x1≤x2≤n,1≤y1≤y2≤m,−1000≤矩阵内元素的值≤1000

输入样例：
3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4

输出样例：
17
27
21

#include <cstdio>
#include <iostream>

using namespace std;

const int N = 1010;

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

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]);
        }
    }
    
    for(int i = 1; i <= n; ++i)
    {
        for(int j = 1; j <= m; ++j)
        {
            s[i][j] = s[i][j-1] + s[i-1][j] + a[i][j] - s[i-1][j-1];  //s[i][j]表示从左上角[1][1]到右下角[i][j]的面积
        }
    }
    
    while(q--)
    {
        int x1,x2,y1,y2;
        scanf("%d%d%d%d", &x1,&y1,&x2,&y2);
        int res = s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1];  //这个画图就可以看出来了
        printf("%d\n", res);
    }
    
    return 0;
}

二、差分

1.概念

这个差分就是前缀和的逆运算，比如 $a[i]=b[1]+b[2]+\cdots +b[i]$ ，那么数组a就是数组b的前缀和数组，数组 $b$ 就为数组 $a$ 的差分数组
那么这个差分有什么用呢？其实就是为了在 $a[l,r]$ 中 $+c$ 用的，使得原来 $O(n)$ 的复杂度变为 $O(1)$
具体就是一个 $insert$ 方法：

void insert(int l, int r, int c)
{
	//a[i] = b[1] + b[2] + ... b[i]
	b[l] += c;  //因为b为a的差分数组，给b[l]+=c那么之后通过数组b来求数组a的话，数组a中的[l,n]每个都会加上c
	b[r+1] -= c;  //数组a中的[r+1,n]每个都会减去c，这样就成了a中[l,r]都加上了c
}

还有就是这个差分矩阵开始不用想着怎么去构造，只需要先把怎么插想好了，然后按照[i,i]插a[i]就可以了

2. 一维差分例题

输入一个长度为 n的整数序列。接下来输入 m 个操作，每个操作包含三个整数 l,r,c，表示将序列中 [l,r] 之间的每个数加上c。请你输出进行完所有操作后的序列。

输入格式
第一行包含两个整数 n和 m。第二行包含 n 个整数，表示整数序列。接下来 m 行，每行包含三个整数 l，r，c，表示一个操作。

输出格式
共一行，包含 n个整数，表示最终序列。

数据范围
1≤n,m≤100000,1≤l≤r≤n,−1000≤c≤1000,−1000≤整数序列中元素的值≤1000

输入样例：
6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1
输出样例：
3 4 5 3 4 2

#include <cstdio>
#include <iostream>

using namespace std;
const int N = 100010;
int a[N], b[N];
int n, m;

void insert(int l, int r, int c)
{
    b[l] += c;
    b[r+1] -= c;
}

int main()
{
    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; ++i) 
    {
        scanf("%d", &a[i]);
        insert(i, i, a[i]);
    }
    
    while(m--)
    {
        int l, r, c;
        scanf("%d%d%d", &l, &r, &c);
        insert(l, r, c);
    }
    
    for(int i = 1; i <= n; ++i) 
    {
        b[i] += b[i-1];  //把数组b直接当成前缀和数组
        printf("%d ", b[i]);
    }
    
    return 0;
}

3. 二维差分例题

输入一个 n行 m列的整数矩阵，再输入 q个操作，每个操作包含五个整数 x1,y1,x2,y2,c，其中 (x1,y1) 和 (x2,y2) 表示一个子矩阵的左上角坐标和右下角坐标。每个操作都要将选中的子矩阵中的每个元素的值加上 c。请你将进行完所有操作后的矩阵输出。

输入格式
第一行包含整数 n,m,q。接下来 n 行，每行包含 m 个整数，表示整数矩阵。接下来 q 行，每行包含 5 个整数 x1,y1,x2,y2,c，表示一个操作。

输出格式
共 n 行，每行 m 个整数，表示所有操作进行完毕后的最终矩阵。

数据范围
1≤n,m≤1000,1≤q≤100000,1≤x1≤x2≤n,1≤y1≤y2≤m,−1000≤c≤1000,−1000≤矩阵内元素的值≤1000

输入样例：
3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1
输出样例：
2 3 4 1
4 3 4 1
2 2 2 2

#include <cstdio>
#include <iostream>

using namespace std;

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

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]);  //刚开始不用担心怎么构造，只用考虑之后的怎么插，开始就跟插一样初始化
        }
    }
    
    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-1][j] + b[i][j-1] - b[i-1][j-1] + b[i][j];
            printf("%d ", b[i][j]);
        }
        puts("");
    }
    
    return 0;
}