蓝桥差分模板

小明的彩灯

题目描述

小明拥有 NN 个彩灯，第 ii 个彩灯的初始亮度为 aiai​。

小明将进行 QQ 次操作，每次操作可选择一段区间，并使区间内彩灯的亮度 +x+x（xx 可能为负数）。

求 QQ 次操作后每个彩灯的亮度（若彩灯亮度为负数则输出 00）。

输入描述

第一行包含两个正整数 N，QN，Q，分别表示彩灯的数量和操作的次数。

第二行包含 NN 个整数，表示彩灯的初始亮度。

接下来 QQ 行每行包含一个操作，格式如下：

l r x，表示将区间 l∼rl∼r 的彩灯的亮度 +x+x。

1≤N,Q≤5×1051≤N,Q≤5×105，0≤ai≤1090≤ai​≤109，1≤l≤r≤N1≤l≤r≤N，−109≤x≤109−109≤x≤109

输出描述

输出共 11 行，包含 NN 个整数，表示每个彩灯的亮度。

输入输出样例

示例 1

输入

5 3
2 2 2 1 5
1 3 3
4 5 5
1 1 -100

输出

0 5 5 6 10

#include <bits/stdc++.h>
using namespace std;

const int MAXN = 500005; 
long long a[MAXN], d[MAXN];

int main() {
    int N, Q;
    cin >> N >> Q;
    
    for (int i = 1; i <= N; ++i) {
        cin >> a[i];
    }

    //差分数组
    memset(d, 0, sizeof(d));

    for (int i = 0; i < Q; ++i) {
        int l, r, x;
        cin >> l >> r >> x;
        //在 d[l] 和 d[r+1] 做固定的操作
        d[l] += x; 
        if (r + 1 <= N) {
            d[r + 1] -= x; 
        }
    }

    long long add = 0;
    for (int i = 1; i <= N; ++i) {
        add += d[i]; //通过差分数组恢复每个位置的增量
        a[i] += add; 
        if (a[i] < 0) a[i] = 0;
    }

    for (int i = 1; i <= N; ++i) {
        cout << a[i] << " ";
    }
    cout << endl;

    return 0;
}