本文共 1327 字,大约阅读时间需要 4 分钟。
给定一个长 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 [ 0 : i ] A[i]=\sum d[0:i] A[i]=∑d[0:i]真实代码中可以想象 A A A一开始都是 0 0 0,每次将 A [ i : i ] A[i:i] A[i:i]增加一个数,而对于差分数组来说,原数组 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越界了那就忽略),这样方便编程。代码如下:
#includeusing namespace std;const int N = 100010;int a[N], d[N];int n, m, l, r, c;int main() { cin >> n >> m; for (int i = 0; i < n; i++) cin >> a[i]; for (int i = 0; i < n; i++) { d[i] += a[i]; if (i + 1 < n) d[i + 1] -= a[i]; } while (m--) { cin >> l >> r >> c; d[l - 1] += c; d[r] -= c; } int s = 0; for (int i = 0; i < n; i++) { s += d[i]; printf("%d ", s); } return 0;}
时间复杂度 O ( n + m ) O(n+m) O(n+m),空间 O ( n ) O(n) O(n)。
转载地址:http://cijs.baihongyu.com/