差分【差分详讲】acwing797
区间操作可能第一时间会想到用线段树之类的,但是可以用差分前缀和,使用起来很方便(我自己写线段树日常出错。。。
暴力的话一次操作的时间复杂度是O(n),但是差分可以O(1)实现。
我们假定输入的这个数组是a1…an,我们构造一个数组,使得a数组是b数组的前缀和:
a1 = b1
a2 = b2 + b1
a3 = b3 + b2 + b1
...
an = bn + ... + b2 + b1
变形:
a1 = b1
a2 = b2 + a1
a3 = b3 + a2
...
an = bn + an-1
推导:
b1 = a1 - a0
b2 = a2 - a1
...
bn = an - an-1
这样的话我们如果需要对a数组的区间[l,r]加上一个数c,那么我们就将b数组的第l个数加上c,这样操作过后再求前缀和的话,就相当于是a数组的l之后的每一个数都增加了c,但是我们不想让r之后的地方加上这个c,所以我们将b数组的第r+1个数减去c,这样就实现了一次操作,时间复杂度是O(1)的。
ac代码:
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
int n,m,l,r,c;
int a[100005],b[100005];
signed main(){
IOS
cin>>n>>m;
for(int i = 1;i <= n;++i)
cin>>a[i],b[i] = a[i] - a[i - 1];
while(m--)
cin>>l>>r>>c,b[l] += c,b[r + 1] -= c;
for(int i = 1;i <= n;++i)
a[i] = b[i] + a[i - 1];
for(int i = 1;i <= n;++i)
cout<<a[i]<<" ";
return 0;
}