例题:acwing797
模板:
#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int a[N],b[N];
int n,m;
void insert(int l,int r,int c){
b[l] += c;
b[r + 1] -= c; //插入函数,任务一:构建差分数组b;任务二:进行数据插入操作
}
int main(){
cin >> n >> m;
for(int i = 1;i <= n;i ++ ){
cin >> a[i];
}
for(int i = 1;i <= n;i ++ ){
insert(i,i,a[i]); //这步是构造差分数组b[i],【注释一】
}
while(m -- ){
int l,r,c;
cin >> l >> r >> c;
insert(l,r,c);
}
for(int i = 1;i <= n;i ++ ){
b[i] += b[i - 1]; //构造前缀和数组b[i]
}
for(int i = 1;i <= n;i ++ ){
cout << b[i] << " ";
}
return 0;
}
差分为前缀和的逆操作。
前缀和中,输入原数组a[i] ,b[i] = b[i - 1] + a[i],求得b[i]为前缀和数组。在差分中,输入的原数组数组a[i]为b[i]的前缀和数组,我们要构造一个b数组,也就是“差分数组”,通过操作差分数组的某一项可以对前缀和数组的某一段进行整体操作。insert函数进行了差分数组的操作,通过b[l] += c;b[r + 1] -= c;可以使前缀和数组a[i]在[l,r]上每一位加c。此外insert数组还完成了“差分数组”b[i]的初始化工作。见注释一。
注释一:
引用acwing797中的数据
insert(1,1,a[1])
b[1] = a[1]
b[2] = -a[1]
insert(2,2,a[2])
b[2] = a[2] - a[1]
b[3] = -a[2]
insert(3,3,a[3])
b[3] =a[3] - a[2]
b[4] = -a[3]
......
insert(6,6,a[6])
b[6] = a[6] - a[5]
b[7] = - a[6]
这样“差分数组”b就构造好了、
下面通过insert函数对b进行操作,并构造前缀和数组b(本模板代码中对a、b数组两个数组没有时时联动关系,对某一数组操作以后需要通过手动操作同步)
模板代码实在优美,就是晦涩难懂。
本文有任何问题欢迎大家指正!