差分

111 阅读1分钟

详解

给定一个数组a1, a2, ......, an,构造一个数组b1, b2, ...... , bn,使得 ai = b1 + b2 + ...... + bi(a是b的前缀和,b是a的差分)。
用处1:给定区间[l, r],要将a数组该区间内的元素都+c,使用差分可以在O(1)时间复杂度内完成。只需要对b[l] + c, b[r + 1] - c,然后对b做前缀和就可以做到这一点。

例题

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

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

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

输入格式

第一行包含两个整数 nn 和 mm。

第二行包含 nn 个整数,表示整数序列。

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

输出格式

共一行,包含 nn 个整数,表示最终序列。

数据范围

1≤n,m≤1000001≤n,m≤100000,
1≤l≤r≤n1≤l≤r≤n,
−1000≤c≤1000−1000≤c≤1000,
−1000≤整数序列中元素的值≤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

代码

#include<iostream>

using namespace std;

const int N = 100010;

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

int main()
{
    cin >> n >> m;
    for (int i = 1; i <= n; i ++ )
        cin >> a[i];    
    for (int i = 1; i <= n; i ++ )
    {
        b[i] += a[i];
        b[i + 1] -= 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 ++ ) cout << b[i] << " ";
    cout << endl;
    return 0;
}