算法基础（五）| 差分算法及模板详解

2022 年 10 月 11 日
山东
本文字数：2257 字
阅读完需：约 7 分钟

⭐写在前面的话：本系列文章旨在复习算法刷题中常用的基础算法与数据结构，配以详细的图例解释，总结相应的代码模板，同时结合例题以达到最佳的学习效果。本专栏面向算法零基础但有一定的 C++基础的学习者。若 C++基础不牢固，可参考：10min快速回顾C++语法，进行语法复习。
🔥本文已收录于算法基础系列专栏：算法基础教程免费订阅，持续更新。

差分

一维差分

差分思想和前缀和是相反的。

首先我们先定义数组 a, 其中 a[1],a[2]...a[n]作为前缀和。

然后构造数组 b，b[1],b[2]...b[n]为差分数组。其中通过差分数组的前缀和来表示 a 数组，即 a[n] = b[1] + b[2]+...+b[n]。

一维差分数组的构造也很简单，即 a[1] = b[1], b[2] = a[2] - a[1], b[n] = a[n] - a[n-1]；

注意：刚开始时可以初始化数组 a,b 全部为 0，输入 a 数组后；在构造时，只需要将 b[1]看做在[1, 1]区间上加上 a[1]; b[2] 看作在[2, 2]区间上加上 a[2]；

//eg:对于b[1]:b[1] = 0 + a[1];b[2] = 0 - a[1];   //最终：b[1] = a[1]//对于b[2]:b[2] = b[2] + a[2];  //==> 最终：b[2] = a[2] - a[1]b[3] = b[3] - b[2];

复制代码

差分数组的好处是可以简化运算，例如想要给一个区间 [l,r] 上的数组加一个常数 c，原始的方法是依次加上 c，这样的时间复杂度是 O(n)的。但是如果采用差分数组的话，可以大大降低时间复杂度到 O(1)。

由于 a[n] = b[1] + b[2]+...+b[n]，因此只需要将 b[l] = b[l] + c 即可，这样 l 之后的数字会依次加上常数 c，而在 b[r]处，将 b[r+1] = b[r+1] - c ，这样 r 之后的数组又会恢复原值，仅需要处理这两个边界的差分数组即可，时间复杂度大大降低。

例题：差分

输入一个长度为 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 31 2 2 1 2 11 3 13 5 11 6 1

复制代码

输出样例：

3 4 5 3 4 2

复制代码

代码模板

#include<iostream>using namespace std;
const int N = 100010;
int m,n;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]);    //插入的方式形成b[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;}

复制代码

二维差分

将 a[i][j]表示为一个差分数列 b[i][j]的和即可。如下所示：

基本思路：给其中的一个子矩阵加上一个值。矩阵以外的减去一个值即可。

可列公式表示各个范围如下：

由上面范围，可以求得最终要算的小正方形的面积公式：

矩阵的初始化；

假定 a[i][j] = 0,b[i][j] =0,然后读取数组 a，只需要对 b 进行插入即可。b[i][j]相当于从(i,j)到(i,j)插入一个 a[i][j]形成的。

最后求 a[i][j]只需要求解 b[i][j]的前缀和即可。

例题：差分矩阵

输入一个 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≤10001≤q≤1000001≤x1≤x2≤n1≤y1≤y2≤m−1000≤c≤1000−1000≤矩阵内元素的值≤1000

输入样例：

3 4 31 2 2 13 2 2 11 1 1 11 1 2 2 11 3 2 3 23 1 3 4 1

复制代码

输出样例：

2 3 4 14 3 4 12 2 2 2

复制代码

代码模板

#include<iostream>using namespace std;
const int N =1010;
int a[N][N],b[N][N];int n, m ,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]);                    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, x2, y1, 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-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;}

复制代码

发布于: 刚刚阅读数: 4

原文链接:【http://xie.infoq.cn/article/ec51f9d5a0755973d06c579e6】。未经作者许可，禁止转载。

timerring

关注

还未添加个人签名 2022.07.14 加入

还未添加个人简介

发布

暂无评论

创作场景

算法基础（五）| 差分算法及模板详解

差分

一维差分

例题：差分

代码模板

二维差分

例题：差分矩阵

代码模板

timerring

评论