输入一个长度为 n 的整数序列。
接下来输入 m 个操作,每个操作包含三个整数 l,r,c,表示将序列中 [l,r] 之间的每个数加上 c。
请你输出进行完所有操作后的序列。
输入格式
第一行包含两个整数 n 和 m。
第二行包含 n 个整数,表示整数序列。
接下来 m 行,每行包含三个整数 l,r,c,表示一个操作。
输出格式
共一行,包含 n 个整数,表示最终序列。
数据范围
1≤n,m≤100000,
1≤l≤r≤n,
−1000≤c≤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
思路
差分可以理解为前缀和的逆运算。
通过暴力直接求解,会超时,然后通过求差分数组,直接对数据进行操作可以把平均时间复杂度降至O(n)。
差分原理:
原数组a:
a
:
a
[
1
]
,
a
[
2
]
,
a
[
3
]
,
.
.
.
,
a
[
n
]
;
a:a[1],a[2], a[3],..., a[n];
a:a[1],a[2],a[3],...,a[n];
构造一个数组b :
b
[
1
]
,
b
[
2
]
,
b
[
3
]
,
.
.
.
,
b
[
i
]
;
b[1] ,b[2] , b[3],..., b[i];
b[1],b[2],b[3],...,b[i];
a
[
i
]
=
b
[
1
]
+
b
[
2
]
+
b
[
3
]
+
.
.
.
+
b
[
i
]
a[i] = b[1] + b[2 ]+ b[3] +... + b[i]
a[i]=b[1]+b[2]+b[3]+...+b[i]
构造差分b数组:
a [ 0 ] = 0 ; a[0 ]= 0; a[0]=0;
b [ 1 ] = a [ 1 ] − a [ 0 ] ; b[1] = a[1] - a[0]; b[1]=a[1]−a[0];
b [ 2 ] = a [ 2 ] − a [ 1 ] ; b[2] = a[2] - a[1]; b[2]=a[2]−a[1];
b [ 3 ] = a [ 3 ] − a [ 2 ] ; b[3] =a [3] - a[2]; b[3]=a[3]−a[2];
. . . ... ...
b [ n ] = a [ n ] − a [ n − 1 ] ; b[n] = a[n] - a[n-1]; b[n]=a[n]−a[n−1];
然后差分数组和元素组有某种关系,我们发现对差分数组b操作,等于对元素组或者说是前缀和数组a进行如下变化:
所以我们可以利用这个性质在平均时间复杂度为O(n)内解决。
代码示例
#include <iostream>
using namespace std;
const int N = 100010;
int n, m;
int a[N], b[N];
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i ++) cin >> a[i];
// 求差分数组b[] a数组是b数组的前缀和数组
for(int i = 1; i <= n; i ++) b[i] = a[i] - a[i-1];
int l, r, c;
while(m --)
{
cin >> l >> r >> c;
b[l] += c;
b[r+1] -= c;
}
// 求前缀和数组
for(int i = 1; i <= n; i ++)
{
a[i] = a[i - 1] + b[i] ;
cout << a[i] << ' ';
}
return 0;
}