题目地址:
https://www.acwing.com/problem/content/799/
给定一个长 n n n的整数序列 A A A,再进行 m m m次操作,每次操作包含三个整数 l , r , c l,r,c l,r,c,表示将序列中的 [ l , r ] [l,r] [l,r]每个数都增加 c c c。输出操作完后的数列。区间的下标是从 1 1 1开始计数的。
输入格式:
第一行包含两个整数
n
n
n和
m
m
m。第二行包含
n
n
n个整数,表示整数序列。接下来
m
m
m行,每行包含三个整数
l
l
l,
r
r
r,
c
c
c,表示一个操作。
输出格式:
共一行,包含
n
n
n个整数,表示最终序列。
数据范围:
1
≤
n
,
m
≤
100000
1\le n,m\le 100000
1≤n,m≤100000
1
≤
l
≤
r
≤
n
1\le l\le r\le n
1≤l≤r≤n
−
1000
≤
c
≤
1000
-1000\le c\le 1000
−1000≤c≤1000
−
1000
≤
A
[
i
]
≤
1000
-1000\le A[i]\le 1000
−1000≤A[i]≤1000
思路是利用差分数组。求出差分数组 d d d使得 A [ i ] = ∑ d [ 1 : i ] , d [ i ] = A [ i ] − A [ i − 1 ] A[i]=\sum d[1:i],d[i]=A[i]-A[i-1] A[i]=∑d[1:i],d[i]=A[i]−A[i−1]对于差分数组来说,原数组 A [ l : r ] A[l:r] A[l:r]全体增加一个数 c c c,效果是 d [ l ] d[l] d[l]增加 c c c,而 d [ r + 1 ] d[r+1] d[r+1]减少 c c c(如果 r + 1 r+1 r+1越界了那就忽略)。代码如下:
#include <iostream>
using namespace std;
const int N = 100010;
int a[N], d[N];
int n, m, l, r, c;
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
d[i] = a[i] - a[i - 1];
}
while (m--) {
scanf("%d%d%d", &l, &r, &c);
d[l] += c;
d[r + 1] -= c;
}
for (int i = 1, s = 0; i <= n; i++) {
s += d[i];
printf("%d ", s);
}
}
时间复杂度 O ( n + m ) O(n+m) O(n+m),空间 O ( n ) O(n) O(n)。