差分就是前缀和的逆运算。跟前缀和类似,给定一些数组a1,a2....an,也给定一些数组b1,b2...bn,其中a[i]=b[i]+a[i-1],也就是说ai是bi的前缀和,反过来bi就是ai的差分数组了。
所以:b[i] = a[i] - a[i - 1]。
题目:
输入一个长度为 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中的数组加r的和。始终要记得,a数组是b数组的前缀和,要求ai+c从l到r的和,也就是把b数组也同时加上r,但是是从b[l]开始加的,一直加到b[r],所以b[r+1]之后的数组要变成b[r+1]-c。画个图说明一下:
![](https://img-blog.csdnimg.cn/img_convert/54c480391b7ebed3f8a7c3cb65021301.png)
题目中求从l到r的a数组加上c的和,所以我们在b数组从b[l]开始全部加上r一直加到最后,因为a[i]=b[i]+a[i-1],也就是说b[i] = a[i] - a[i - 1],所以在b[r+1]之后的元素要减去c,减去c的元素就是原来普通的b元素的值。
因此我们得出一维差分结论:给a数组中的[ l, r]区间中的每一个数都加上c,只需对差分数组b做
b[l] + = c, b[r+1] - = c
第一步,创建差分数组b[i]=a[i]-a[i-1],第二步,b[l] + = c, b[r+1] - = c,第三步,a[i]=b[i]+a[i-1].
代码实现:
#include<iostream>
using namespace std;
const int N=100010;
int a[N],b[N];
int main(){
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++){
cin>>a[i];
b[i]=a[i]-a[i-1];//第一步,创建差分数组b[i]=a[i]-a[i-1]
}
int l,r,c;
while(m--){
cin>>l>>r>>c;
b[l]+=c;
b[r+1]-=c;//第二步,写出b[l]和b[r+1]表达式
}
for(int i=1;i<=n;i++){
a[i] = b[i] + a[i - 1]; //第三步,写出a[i]的表达式
cout<<a[i]<<" ";
}
return 0;
}
二维差分:
跟一维差分一样,给出了a[ij]数组,其中b[ij]数组为a[ij]数组的差分数组,a[ij]数组为b[ij]数组的前缀和数组。类比一维差分,我们给出了以下操作使之成为二维差分:
b[x1][y1] + = c;相当于从左上角边界一直到右下角的整个矩形都加上c
b[x1,][y2+1] - = c;相当于图中绿色的部分全部减去c,得到原本的b[ij]的值
b[x2+1][y1] - = c;相当于图中红色的部分全部减去c,得到原本的b[ij]的值
b[x2+1][y2+1] + = c;从图中可以看到,一开始从左边界到右下角的数组元素全部被加上c,但是被红色和绿色都剪去了c,也就是加了一个c减去了两个c,所以还要加上一个c
![](https://img-blog.csdnimg.cn/img_convert/9f46813d4783b074105858043262da17.png)
我们将上述操作构造一个insert插入函数:
void insert(int x1,int y1,int x2,int y2,int c)
{ //对b数组执行插入操作,等价于对a数组中的(x1,y1)到(x2,y2)之间的元素都加上了c
b[x1][y1]+=c;
b[x2+1][y1]-=c;
b[x1][y2+1]-=c;
b[x2+1][y2+1]+=c;
}
我们可以先假想a数组为空,那么b数组一开始也为空,但是实际上a数组并不为空,因此我们每次让b数组以(i,j)为左上角到以(i,j)为右下角面积内元素(其实就是一个小方格的面积)去插入 c=a[i][j],等价于原数组a中(i,j) 到(i,j)范围内 加上了 a[i][j] ,因此执行n*m次插入操作,就成功构建了差分b数组。
代码如下:
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
insert(i,j,i,j,a[i][j]); //构建差分数组
}
}
题目:
输入一个 n行 m 列的整数矩阵,再输入 q 个操作,每个操作包含五个整数 x1,y1,x2,y2,其中 (x1,y1) 和 (x2,y2)表示一个子矩阵的左上角坐标和右下角坐标。
每个操作都要将选中的子矩阵中的每个元素的值加上 c
请你将进行完所有操作后的矩阵输出。
输入格式
第一行包含整数 n,m,q
接下来 n 行,每行包含 m个整数,表示整数矩阵。
接下来 q 行,每行包含 5 个整数 x1,y1,x2,y2表示一个操作。
输出格式
共 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<iostream>
#include<dstdio>
using namespace std;
const int N = 100010;
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()
{
int n, m, q;
cin >> n >> m >> q;
for (int i = 1; i <= n; i++)
for (int j = 1; j <= m; j++)
cin >> 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 - 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]);
}
printf("\n");
}
return 0;
}
总的来说一维差分的核心是:第一步,创建差分数组b[i]=a[i]-a[i-1],第二步,b[l] + = c, b[r+1] - = c,第三步,a[i]=b[i]+a[i-1].
二维差分的核心就是上面的insert函数。