差分acwing797. 差分
设前缀数组为:a
差分数组为:b
a为b的前缀数组,b为a的差分数组
使得a[i] = b[1] + b[2 ]+ b[3] +, + b[i]
如下:
a[0 ]= 0;
b[1] = a[1] - a[0];
b[2] = a[2] - a[1];
b[3] =a [3] - a[2];
…
b[n] = a[n] - a[n-1];
给一个区间[l,r],a数组的这个区间的每一个数加C。
a[l] + c , a[l+1] + c , a[l+2] + c , a[r] + c ;
始终要记得,a数组是b数组的前缀和数组,比如对b数组的b[i]的修改,会影响到a数组中从a[i]及往后的每一个数。
首先让差分b数组中的 b[l] + c ,a数组变成 a[l] + c ,a[l+1] + c, a[n] + c;
然后我们打个补丁,b[r+1] - c, a数组变成 a[r+1] - c,a[r+2] - c,a[n] - c;
一维差分结论:给a数组中的[ l, r]区间中的每一个数都加上c,只需对差分数组b做 b[l] + = c, b[r+1] - = c。时间复杂度为O(1), 大大提高了效率。
#include <iostream>
using namespace std;
const int N = 1e5 + 5;
int a[N],b[N];//a为前缀和数值,b为差分数组
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];
}
while(m-->0){
int l,r,c;
cin >> l >> r >> c;
b[l]+=c;
b[r+1]-=c;
}
for(int i=1;i<=n;i++){
a[i]=a[i-1]+b[i];
cout << a[i] << " ";
}
return 0;
}