前缀和与差分（c++）

前缀和

基本思想

原数组：a1 a2 a3 …an
前缀和：si = a1 + a2 + a3 + …+ ai (s[0] = 0 例如：[1, 10] s = s10 - s0)
1.求si for(int i = 1; i <= n; i++) s[i] = s[i - 1] + a[i];
2.作用：快速求出原数组中一段的和 例如：[l , r] s = sr - sl;

基本模板
来自yxc

/*一维前缀和 —— 模板题 AcWing 795. 前缀和*/
S[i] = a[1] + a[2] + ... a[i]
a[l] + ... + a[r] = S[r] - S[l - 1]

/*二维前缀和 —— 模板题 AcWing 796. 子矩阵的和*/
S[i, j] = 第i行j列格子左上部分所有元素的和
以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵的和为：
S[x2, y2] - S[x1 - 1, y2] - S[x2, y1 - 1] + S[x1 - 1, y1 - 1]

经典例题

ACWing. 795.前缀和（一维）

输入一个长度为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

解答：（注意s[0] == 0）

#include<iostream>
using namespace std;
const int N = 100010;
int n,m;
int a[N], s[N];
int main()
{
  
    scanf("%d%d",&n, &m );
    for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
    for(int i = 1; i <= n; i++) s[i] = s[i - 1] + a[i];//初始化前缀和数组
    while(m --)
    {
        int l, r;
        scanf("%d%d", &l, &r);//输入区间
        printf("%d\n", s[r] - s[l - 1]);//求[l, r]这一段的和
        
    }
    return 0;
}

ACWing. 796.子矩阵的和（二维）
输入一个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

解答
s[i, j] 表示（i，j）点矩阵坐上角所有元素的和
1、计算 s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
2、算（x1, y1）(x2, y2) 这一子矩阵的和

#include<iostream>
const int N = 1010;
int n, m, q;
int a[N][N], 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", &a[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] + a[i][j];
        
        //询问   
    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;
}

差分

基本思想

原数组：a1 a2 a3 …an
构造数组：b1 b2 b3…bn
使得 ai = b1 + b2 + b3 + …+bi
(一维）
b1 = a1
b2 = a2 - a1
.
.
bn = an - an-1

基本模板 来自yxc

/*一维差分 —— 模板题 AcWing 797. 差分*/
给区间[l, r]中的每个数加上c：B[l] += c, B[r + 1] -= c

/*二维差分 —— 模板题 AcWing 798. 差分矩阵*/
给以(x1, y1)为左上角，(x2, y2)为右下角的子矩阵中的所有元素加上c：
S[x1, y1] += c, S[x2 + 1, y1] -= c, S[x1, y2 + 1] -= c, S[x2 + 1, y2 + 1] += c

经典例题

ACWing797.差分（一维）
输入一个长度为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

解答
将序列中[l, r]之间的每个数加上c
al + c , al+1 + c, …, ar + c;
因为ai = b1 + b2 + …+ bi 所以bi + c ai~an 中每个元素都+c
bl + c , al以及后面的元素都+c
br+1 - c, ar+1 后面的元素都-c
因此最后仅[l, r]区间的a 数组元素+c

#include<iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], b[N];//a[N]为原序列  b[N]为差分序列
void insert(int l, int r, int c)//在[l, r] 之间的元素都＋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]);
    for(int i = 1; i <= n; i++) insert(i , i , a[i]);//这一步是改变b，使得b仍然是a的差分数组

    //询问
    while(m--)//这一步是主要操作，通过操作差分数组b，间接使得把a数组【l,r】里面的数全部加上c
    {
        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】就变成了a【i】   ;对差分序列进行求和来获得原序列,∵差分序列的前缀和=原序列
    b[i] += b[i - 1];//s[i]=s[i-1]+a[i]
    for(int i = 1; i <= n; i++) printf("%d ", b[i]);
    return 0;
}

ACWing798.差分矩阵（二维）
输入一个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

解答
原：aij
差分：bij
使得：bij 左上角的所有元素和等于aij
给子矩阵加上一个值
bxy + c 为a数组右下角矩阵元素都+c

#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]);
    
    for(int i = 1; i <= n; i++)
       for(int j = 1; j <= m; j++)
           insert(i, j, i, j, a[i][j]);//先插入a[i][j]
    
    while(q--)
    {
        int x1, y1, x2, y2, c;
        cin >> x1 >> y1 >> x2 >> y2 >> c;
        insert(x1, y1, x2, y2, c);//插入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];//求前缀和
    
     for(int i = 1; i <= n; i++)
     {
        for(int j = 1; j <= m; j++)  printf("%d ", b[i][j]);
        puts("");
    
     }
     return 0;
}