前缀和与差分

前缀和

原数组：a1 a2 a3 a4 a5 a6 …

前缀和：Si = a1 + a2 + a3 +… + ai

前缀和实际上就是数组中一段区间的和

前缀和作用就是快速算出一段区间的和

问题：为什么一定要从1开始，因为我们可以预留出来0的位置

因为求某一段区间的和([l,r])

公式为：Sl - Sr-1

如果求[1,10]，则需要S10 - S0，所以预留出来0的位置可以统一算法

一维前缀和

代码

#include <iostream>
using namespace std;

const int N = 100010;
int a[N],s[N];
int n,m;

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];
    
    int l,r;
    for(int i = 0;i < m; i ++){
        scanf("%d%d",&l,&r);
        printf("%d\n",s[r] - s[l - 1]);
    }
    
    return 0;
}

矩阵中的前缀和

差分

差分其实和前缀和是逆运算

例如：b1,b2,b3,b4…bn

那么ai = b1 + b2 + b3 + … + bi，那么ai组成的序列为B的前缀和，B则成为A的差分

一维差分的应用

典型例子就是给定一组序列，求某范围内所有的元素加上某一个相同的值

可以利用差分数组的左边界+c，右边界右一位-c，这样根据这个差分求前缀和，范围内都加上了c

达到了目的

---

典型例题：

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

题目给出了前缀和数组，不必特意去构建差分数组

可以利用插入更新的方式，假设A前缀和数组元素都为0，那么显然B差分数组的元素都为0，此时就构建了一个差分数组，可以利用插入的方式，那么只需要每次在[1,1]区间内插入a1，即使A数组的第一个元素成为了a1,同时维护了差分数组和前缀和数组，后面的元素以此类推

代码：

#include <iostream>
using namespace std;

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

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]);
    
    for(int i = 1;i <= n;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];
    
    for(int i = 1;i <= n;i ++) printf("%d ",b[i]);
    
    return 0;
}

矩阵的差分

矩阵的差分与二维矩阵求前缀和也是一组逆运算

矩阵的差分与一维类似，不同的是，矩阵的插入操作，即子矩阵每个元素加上某值，注意打补丁时要加回一个小矩阵

然后构建差分矩阵也和一维差分类似，用更新的方式构建

代码：

#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[x1][y2 + 1] -= c;
     b[x2 + 1][y1] -= 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]);
        
    while(q --){
        int x1,y1,x2,y2,c;
        cin >> 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 - 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;
}