一维矩阵的差分

一维矩阵差分

首先明确一点，差分是前缀和的逆运算，即要求一个数组a的差分数组b，就是要使得b数组的前缀和数组是a数组。那么如何构造差分数组呢？我们可以这样看，a[1]=b[1],a[2]=b[1]+b[2],a[3]=b[1]+b[2]+b[3],…a[n]=b[1]+b[2]+…+b[n],那么b[1]=a[1],b[2]=a[2]-b[1]=a[2]-a[1],b[3]=a[3]-b[1]-b[2]=a[3]-a[2],…,b[n]=a[n]-b[1]-b[2]-…-b[n - 1]=a[n]-a[n-1]，通过以上的推导，我们可以很清楚的看出如何推导差分矩阵，b[n]=a[n]-a[n-1];但是在使用代码实现的时候，我们并不需要这样做，我们可以假设差分数组b初始全为0，那么为了使其前缀和为a数组，我们想办法往b数组中插入数据，假设我们在b[i]处插入a[i]，由于a[n]=b[1]+b[2]+b[3]+…+b[n]，所以如果给b[i]加上a[i]，那么a[i],a[i+1],…a[n]都会由于此操作而加上a[i]，但是这样做就改变了原数组，所以我们在b[i+1]处减去a[i]就好了，这样从a[i+1]开始到a[n]就都会减去a[i]，就没什么变化了。差分解决的问题如下：

输入一个长度为 n 的整数序列。

接下来输入 m 个操作，每个操作包含三个整数 l,r,c，表示将序列中 [l,r] 之间的每个数加上 c。

请你输出进行完所有操作后的序列。

如果我们暴力操作，那么需要使用循环将a[l]到a[r]的所有数据加上c，这样的时间复杂度是o(n)的，但是使用差分，我们可以使的复杂度降低到o(1)，我们将要改变的数组看做原数组a，那么在[l.r]区间的每一个数加上c，就可以通过其差分数组b,在b[l]处加c，这样会使得a[l]及之后的所有数都加上c（具体为什么这样，可以想一下前缀和是如何求的），接下来我们再将b[r+1]减c，这样a[r+1]及之后的所有数都会减去c，这样操作后，就使得a[l]到a[r]中的所有数加上了c。那么根据这个原理，在构造差分矩阵的时候就可以在b[i]处加a[i],在b[i+1]处减a[i]即可（其实就是刚开始的意思，不过可能前面说的不是很清楚）。接下来看一下代码实现：

#include<iostream>
#include<algorithm>

using namespace std;

const int N = 100010;

int n, m;
int a[N], b[N];
int l, r, c;

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

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]);//构造差分数组
    
    while(m --)
    {
        cin >> l >> r >> c;
        insert(l, r, c);
    }
    
    for(int i = 1; i <= n; i ++) b[i] += b[i - 1];
    
    for(int i = 1; i <= n; i ++) cout << b[i] << " ";
    
    return 0;
}