题意
TT 有一只猫 ,它从 世界地图 选了 n 个城市,用 ai 表示每个城市的资产 。
猫会给出几个操作, 区间 [ l , r ] 的城市资产都加 c 。在q次操作后,输出所有城市的资产。
Input
第一行有两个数 n, q (1<=n,q<=2*10^5) ,n表示城市的个数,q表示操作次数。
第二行包含n个数,分别代表每个城市的资产ai。(-10^6 <=ai<=10^6)
接下来q行,每一行代表一个操作, l ,r ,c
(1≤l≤r≤n,−10^5 ≤c≤ 10^5)(1≤ l ≤ r ≤n,−10^5 ≤ c ≤ 10^5)
output
每个城市最后的资产
Examples
解题思路
1.拒绝暴力求解,因为数据太大,肯定超时。
2.利用差分的方法。
假设原数组为A,差分数组为B,范围【1,n】
那么 B[1] = A[1]
B[i] = A[i] - A[i-1]
差分的性质
1.SUM{ B[1~i] } = A[i] //累加B[ 1~ i ] = A[i]
2.A[L] ~ A[R] 均加上c 等价于 B[L] += c,B[R+1] -=c
关于第2 点的解释理解: B[L] 加上 c之后,A[L] =SUM{B[1~L]} ,我们可以知道 ,从L开始,往后的A 都会加上c (因为 B[L] 加了c)。
然后在 B[R+1] 减 c ,是为了和 B[L] 加的c抵消,从 B[R]往后,A 的值都会加c减c,所以不变。
即,操作了B,就相当于操作了一堆A,这显然能够提高我们的速度。
代码实现
#include<cstdio>
const int Max = 200100;
long long a[Max];
long long l , r , n , q , c;
long long b[Max];
int main()
{
scanf("%lld %lld",&n,&q);
for( long long i =1 ; i<=n ;i++) scanf("%lld",&a[i]);; //1到n
//预处理
for( long long i=1 ; i<=n ; i++) b[i] = a[i] - a[i-1];
// for( int i=1;i<=n;i++)
// cout<<b[i]<<" ";
for ( long long i=1 ; i <= q; i++)
{
scanf("%lld %lld %lld",&l,&r,&c);
b[l] +=c;
b[r+1] -= c; //差分
}
// for( int i=1;i<=n;i++)
// cout<<b[i]<<" ";
printf("%lld",b[1]);
long long sum = b[1];
for(long long i=2;i<=n;i++)
{
sum +=b[i];
printf(" %lld",sum);
}
return 0;
}
小结
差分确实是一个非常好的方法,把原本的暴力 O(qn)降到了O(n+q)