【数据结构与算法】差分---前缀和的逆运算

 数据结构与算法之差分 --- 前缀和的逆运算

目录

什么是差分

构造差分数组

差分数组的应用

经典例题

来源：797. 差分 - AcWing题库

---

什么是差分

        差分（difference）又名差分函数或差分运算，差分的结果反映了离散量之间的一种变化，差分是前缀和的逆运算。差分有分阶，常见的是一阶差分。

构造差分数组

给定一个数列a[n]:a1,a2,a3,a4,a5,...an;

构造一个数列b[n]:b1,b2,b3,b4,b5,...bn;

使得ai=b1+b2+b3+...+bi;

b就称为a的差分，a称为b的前缀和。

差分数组的应用

1.a和b的求解

已知ai=b1+b2+b3+...+bi，可以推导出：

        b1=a1;

        b2=a2-a1;

        b3=a3-a2;

        ...

        bn=an-a(n-1);

那么，当a已知时可以通过a求b：bn=an-a(n-1);

反过来，当b已知时可以通过b求a：an=bn+a(n-1);

2.对a的随机区间进行快速更改

假设题目要求数组a[n]序列中 [l,r]之间的每个数加上 c，一共进行m次这样的处理。

方法一：暴力破解

        写一个for循环， 范围为l-r，a[l] + c , a[l+1] + c , a[l+2] + c ...a[r] + c;时间复杂度是O(n)。题干要求要对a[n]进行m次这样的处理，那么时间复杂度就变成了O(n*m)。特别费时。

方法二：利用差分       

        如果b[l]+c，a会有什么变化呢：

        a[1]=b1;                                               无变化

        ...                                                         无变化

        a[l-1]=b1+b2+...+b(l-1);                       无变化

        --------------------------------------------------------------------------------------

        a[l]=b1+b2+...+(bl+c);                                                    变大了c

        a[l+1]=b1+b2+...+(bl+c)+b(l+1);                                     变大了c

        ...                                                                                    变大了c

        a[l+1]=b1+b2+...+(bl+c)+b(l+1)+...+bn;                          变大了c

        -------------------------------------------------------------------------------------

        很显然,[l,n]区间的a都实现了+c，但是r<=n，还没有达到题目效果，所以我们需要进一步处理一下：将b[r+1] - c,从 b[l]+c的表现来看，不难得出效果为[r+1,n]区间的a都实现了-c,这就和上面多出来的[r+1,n]区间的a都+c效果抵消掉了，正好实现题目的要求。

        在这个过程中，时间复杂度为O(1),整体时间复杂度降为O(1*m)，效率大大提高。由此可以总结出：给a数组中的[ l, r]区间中的每一个元素数值都加上c,只需对数组b进行 b[l] + = c, b[r+1] - = c操作即可。

经典例题

来源：797. 差分 - AcWing题库

题目描述 

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

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

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

输入格式

第一行包含两个整数 n和 m。

第二行包含 n 个整数，表示整数序列。

接下来 m 行，每行包含三个整数 l，r，c，表示一个操作。

输出格式

共一行，包含 n个整数，表示最终序列。

数据范围

1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤1000,
−1000≤整数序列中元素的值≤1000

输入样例：

6 3
1 2 2 1 2 1
1 3 1
3 5 1
1 6 1

输出样例：

3 4 5 3 4 2

 参考代码

//首先得明确一点：差分是前缀和的逆运算
//当a已知时可以通过a求b：bn=an-a(n-1);当b已知时可以通过b求a：an=bn+a(n-1);
//差分模板b[l]+=c;b[r+1]-=c;

#include<iostream>
#include<algorithm>
using namespace std;

const int N=100010;
int n,m;
int a[N],b[N];

//差分模板 
void insert(int l,int r,int c)
{
    b[l]+=c;
    b[r+1]-=c;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    
    //法一 a已知 通过a求b：bn=an-a(n-1);
    for(int i=1;i<=n;i++)  b[i]=a[i]-a[i-1];  //可以合并到读取部分，多写一次循环是为了和法二区分
    
    while(m--)
    {
        int l,r,c;
        scanf("%d%d%d",&l,&r,&c);
        insert(l,r,c);
    }
    
    for(int i=1;i<=n;i++) a[i]=b[i]+a[i-1];
    
    for(int i=1;i<=n;i++) printf("%d ",a[i]);
    
    /*
    //法二 直接先把a的值存进b里去
    for(int i=1;i<=n;i++) insert(i,i,a[i]); 
    
    while(m--)
    {
        int l,r,c;
        scanf("%d%d%d",&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 ++ ) printf("%d ", b[i]);
    */
    return 0;
}