差分

题目描述

输入一个长度为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[] ={3,2,5,8,4,7,6,9};的数组中一段加上一个数
我们可以构造一个差分数组b[N];最后在对b[N]求上自己的前缀和就可以得到答案

差分构造

a[i] = b[i] + b[i+1] +…+b[n];

核心操作

对于一段区间[l,r],如果我们想对于a数组区间中的所有数都加上一个数那么就是这个操作

b[l] += c; 1
b[r+1] -=c; 2

第1个式子表示从l下标开始到最后，每个数加上c
第2个式子表示从r+1下标开始到最后，每个数都减去c
所以这两个式子就是对于[l,r]上的数都加了c

差分还原前缀

对于a数组是b数组的前缀，要求出a数组[l,r]上+c后的数组a，b是a加上c后的差分，由b还原a，就是对b[i]求前缀和
b[i] += b[i-1];

代码实现

#include<iostream>
using namespace std;
const int N =100010;

int a[N],b[N];
int n,m;

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[N]初始化b[N]
    for(int i=1;i<=n;++i) insert(i,i,a[i]);
    
    while(m--)
    {
        int l,r,c;
        cin>>l>>r>>c;
        //构造差分数组
        insert(l,r,c);
    }
    //差分数组的逆数组前缀和数组就是答案
    for(int i=1;i<=n;++i) a[i] = a[i-1] + b[i];
    for(int i=1;i<=n;++i) cout<<a[i]<<" ";
    return 0;
}