前缀和与差分

前缀和与差分

前缀和与差分互为逆运算

背景

考虑这样的情况：

给定一组操作，每个操作是对区间[l,r]中的每个元素加上一个值val

如果一个一个地加，时间复杂度将是O(n²)

使用差分数组可以降到O(n)

原理

现在我们定义数组a为数组b的前缀和数组

则数组a满足：a[i] = b[1]+b[2]+...+b[i] = a[i-1] + b[i]

相对应的，此时数组b为数组a的差分数组

由前缀和数组a的性质 则b数组满足：b[i] = a[i] - a[i-1]

差分数组的性质： 结合背景，假如要对区间[l,r]上的元素加上一个值val 原数组是a，现在我们得到了差分数组b,我们可以很轻易地使用差分数组b运算得到a（因为a是b的前缀和数组）

我们先考虑b[l] 如果给b[l]加上val,当我们用b数组恢复a数组时：

由于a[i] = a[i-1] + b[i]

a[l] = b[1]+b[2]+...+b[l];

a[l+1] = b[1]+b[2]+...+b[l]+b[l+1];

注意：此时的b[l]是原来的b[l]加上了val，b数组的其他项不变

我们可以观察到，从a[l]开始，在计算前缀和的过程中，后面所有的项都加上了val 而我们需要加到l截止，所以再给b[r+1]减去val

效果如下：

a[l] = b[1]+b[2]+...+b[l];

a[l+1] = b[1]+b[2]+...+b[l]+b[l+1];

a[r] = b[1]+b[2]+...+b[l]+...+b[r];

a[r+1] = b[1]+b[2]+...+b[l]+...+b[r]+b[r+1];

从a[r+1]开始，后面的项中包含b[l]与b[r+1]，而这两项一个加上了val，一个减去了val，值不变

至此，我们实现了对区间[l,r]上的每一个元素加上val的操作

代码

需求：

给一个长度为n的序列，m次询问

每次询问给一个区间[l,r]和值c，要求将第l个数到第r个数全部加上c

#include <iostream>
using namespace std;
const int N = 1e5+10;
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];
    while(m--){
        int l,r,c;
        cin >> l >> r >> c;
        b[l] += c;
        b[r+1] -= c;
    }
    for(int i = 1; i <= n; i++) b[i] += b[i-1];
    for(int i=1 ; i<=n ; i++) a[i] += b[i];
    for(int i = 1; i <= n; i++) cout << a[i] << ' ';
    return 0;
}