题目描述:
长度为n的数组,一共q次操作,1<=n,q<=1e5;每次操作给出L,R,c,表示区间[L, R]中各个数均加上c,求q次操作结束后,数组中各个元素值。
Input:
第一行有两个值,n,q;
第二行包含n个值,表示数组中的每个值;
接下来q行,每行有 L R c三个值,表示一次操作。
sample input:
4 2
-3 6 8 4
4 4 -2
3 3 1
Output:
依次输出经过所有操作后的数组元素。
sample output:
-3 6 9 2
个人思路:
每次操作遍历一遍数组,加上c,这样的时间复杂度是O(n*p)的,太高。
想一下差分以及前缀和,因为对于差分数组来说,原数组的区间操作等价于差分数组的单点操作,
例如:
A[i] = B[i] - B[i+1],A数组的区间操作等价于B数组的单点操作,
B[L] ~ B[R]均加上c等价于A[L] += c,A[R+1] -= c;
Why?(以下内容强烈建议动手写一写,回顾高中的知识。。)
令:
A[1] = B[1] - 0;
A[2] = B[2] - B[1];
A[3] = B[3] - B[2];
···
A[i] = B[i] - B[i-1];
根据上式计算
B[R] = A[R] + A[R-1] + ··· + A[1];
B[R-1] = A[R-1] + A[R-2] + ··· + A[1];
...
B[L+1] = A[L+1] + A[L] + A[L-1] + ··· +A[1];
B[L] = A[L] + A[L-1] + ··· +A[1];
简单的推导,不难看出,对于B[j]的操作,等价于对其中的单点操作。
那么根据推导的公式,仅需要线性的时间复杂度就完成了。
另外,还有一点,观察上述描述:
有 B[i] = B[i-1] + A[i]
利用其可以保证输出时是线性的。
代码实现:
#include<iostream>
using namespace std;
int n, q;
long long B[200020];
int main() {
scanf_s("%d%d", &n, &q);
int l, r, c;
int num[2];
scanf_s("%d", &num[1]);
B[1] = num[1];
for (int i = 2; i <= n; ++i) {
int s = i % 2;
scanf_s("%d", &num[s]);
if (s == 0)
B[i] = num[0] - num[1];
else
B[i] = num[1] - num[0];
}
for (int i = 1; i <= q; ++i) {
scanf_s("%d%d%d", &l, &r, &c);
B[l] += c;
B[r+1] -= c;
}
long long city = 0;
for (int i = 1; i < n; ++i) {
city += B[i];
printf("%lld ", city);
}
city += B[n];
printf("%lld\n", city);
return 0;
}