【C++】一维、二维差分+模板+例题

引语

前缀和那篇看了吗？快速跳转！

今天来了解下前缀和的逆运算——差分！

PS：此文中含有一个角色小A，我将会努力插入这个名为"小A"的角色来丰富文章趣味

正文

差分的定义

差分可以简单的看成序列中每个元素与其前一个元素的差。

差分主要用于让一个序列某一特定范围内的所有值都加上或减去一个常数。

二维差分是指对于一个n*m的矩阵a，要求支持操作pro(x1,y1,x2,y2,a)，表示对于以(x1,y1)为左上角，(x2,y2)为右下角的矩形区域，每个元素都加上常数a。求修改后的矩阵a。

我们一般将差分看作前缀和的逆运算

一维差分

例题引入

AcWing 797

思路1

小A:暴力YYDS

旁白:小A写好了代码

#include <bits/stdc++.h>
using namespace std;
int n,m,a[100005];
int main(){
	cin >> n >> m;
	for (int i = 1;i <= n;i++) cin >> a[i];
	while (m--){
		int l,r,c;
		cin >> l >> r >> c;
		for (int i = l;i <= r;i++) a[i] += c;
	}
	for (int i = 1;i <= n;i++) cout << a[i] << " ";
	return 0;
}

 旁白:不出意料，小ATLE了

小A:😟

思路2

差分算法，我们需要额外使用一个数字来存储每个数字与上一个数字的差，我这里命名为数组s

我们的数组下标从1开始算
s[i] = a[i] - a[i-1];

那么求一下数组s的前缀和，根据上面公式，我们可以知道：

  s[1]+s[2]+s[3]+s[4]+...+s[i]

=(a[1]-a[0]（a[0]是0）)+(a[2]-a[1])+(a[3]-a[2])+(a[4]-a[3])+...+(a[i]-a[i-1])

=a[1]+(a[2]-a[1])+(a[3]-a[2])+(a[4]-a[3])+...+(a[i]-a[i-1])
 =a[i]
 所以，a[i]=s[1]+s[2]+...+s[i]

小A:我知道了！那我们应该怎么使用差分呢？

给区间[l,r]每个数加上c是:s[l] += c,s[r + 1] -= c

附加:减去是s[l] -= c,s[r + 1] += c

小A:我写个函数吧!

旁白:小A写好了函数

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

那么既然我们前面知道了a[i]=s[1]+s[2]+...+s[i]，那么应该怎么算

for (int i = 1;i <= n;i++){
	int sum = 0;
	for (int j = 1;j <= i;j++){
		sum += s[j];
	}
	cout << sum << " ";
}

其实还可以更简便，直接在s数组上算，直接遍历s数组，然后...

小A:我明白了！s[1]+=s[0],s[2]+=s[1]，这样s[i]+=s[i-1]，算的就是累加和了

for (int i = 1;i <= n;i++){
    s[i] += s[i - 1];
    cout << s[i] << " ";
}

旁白:小A AC 了这道题

总结模板

// 给区间[l,r]每个数加上c是:S[l] += c;S[r + 1] -= c
#include <bits/stdc++.h>
using namespace std;
int a[100005],s[100005],n,m,L,R,C;
void insert_num(int l,int r,int c){
    s[l] += c,s[r + 1] -= c;
    // 减去:s[l] -= c,s[r + 1] += c;
}
int main(){
    cin >> n >> m;
    for (int i = 1;i <= n;i++){
        cin >> a[i];
        insert_num(i,i,a[i]);
    }
    while (m--){
        cin >> L >> R >> C;
        insert_num(L,R,C);
    }
    for (int i = 1;i <= n;i++){
        s[i] += s[i - 1];
        cout << s[i] << " ";
    }
    return 0;
}

二维差分

例题引入

AcWing 798

思路1

小A:我知道了，不能暴力，但是我还是要暴力

旁白:小A写好了代码

#include <bits/stdc++.h>
using namespace std;
int a[1005][1005],n,m,q;
int main(){
    cin >> n >> m >> q;
    for (int i = 1;i <= n;i++){
        for (int j = 1;j <= m;j++){
            cin >> a[i][j];
        }
    }
    while (q--){
        int x1,y1,x2,y2,c;
        cin >> x1 >> y1 >> x2 >> y2 >> c;
        for (int i = x1;i <= x2;i++){
        	for (int j = y1;j <= y2;j++){
        		a[i][j] += c;
			}
		}
    }
    for (int i = 1;i <= n;i++){
        for (int j = 1;j <= m;j++){
            cout << a[i][j] << " ";
        }
        cout << endl;
    }
    return 0;
}

思路2

既然有一维差分，那就有二维差分啦

在(x1,y1)-(x2,y2)区域内，差分影响的是整个黄色区域，多影响两个蓝色部分。绿色区域是两个蓝色区域重叠的部分，所以要加上与一块蓝色区域操作的反操作（即补上）。

让左上(x1,y1)到右下(x2,y2)区域内所有元素加上c

附加：让左上(x1,y1)到右下(x2,y2)区域内所有元素减上c

通过s数组计算值

小A:写代码！

旁白:小A AC 了

总结模板

#include <bits/stdc++.h>
using namespace std;
int a[1005][1005],s[1005][1005],n,m,q;
void insert_num(int x1,int y1,int x2,int y2,int c){
    s[x1][y1] += c,s[x2 + 1][y1] -= c,s[x1][y2 + 1] -= c,s[x2 + 1][y2 + 1] += c;
    // 减去:s[x1][y1] -= c,s[x2 + 1][y1] += c,s[x1][y2 + 1] += c,s[x2 + 1][y2 + 1] -= c;
}
int main(){
    cin >> n >> m >> q;
    for (int i = 1;i <= n;i++){
        for (int j = 1;j <= m;j++){
            cin >> a[i][j];
            insert_num(i,j,i,j,a[i][j]);
        }
    }
    while (q--){
        int x1,y1,x2,y2,c;
        cin >> x1 >> y1 >> x2 >> y2 >> c;
        insert_num(x1,y1,x2,y2,c);
    }
    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];
            cout << s[i][j] << " ";
        }
        cout << endl;
    }
    return 0;
}

结语

好好复习！

参考资料

1 2 3