输入一个长度为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
思路分析
给定a[1],a[2],…a[n]构造差分数组b[n],使得
a[i]=b[1]+b[2]+…+b[i];
当想让a[L~R]中的每个数全部加上一个数V时,等价于b[L]+=V,b[R+1]-=V;
a[1~L-1]=b[1]+b[2]+…+b[L-1];a[1 ~L-1]无影响
a[L~R]=b[1]+b[2]+…+b [ L ](此时的b[ L ]加上了V)+…;所以等价于a[L ~R]中的每个数都加上了V。
a[R+1~ N]=b[1]+b[2]+…+b[ L](此时的b[ L]加上了V)+…+b[ R+1](此时的b[ R+1]减去了V,两者抵消)+…;所以a[R+1~N]无影响。
初始时,b[n]的每个元素都为0,然后以差分操作构造差分数组。
代码
代码1 将O(n)的操作转成O(1)的插入
#include <iostream>
using namespace std;
int a[100005],b[100005];
int n,m;
//对差分数组进行插入操作
void insert(int l,int r,int v){
b[l]+=v;//a[l~r]的数都相当于加上了一个数v,其他数无影响
b[r+1]-=v;
}
int main()
{
int l,r,t;
cin>>n>>m;
//相当于从全部是0的数组开始构造差分数组
//注意点:不能从0开始存储,题目下标从1,不然会导致错误
for(int i=1;i<=n;i++){
cin>>a[i];
insert(i,i,a[i]);
}
while(m--){//进行m次差分操作
cin>>l>>r>>t;
insert(l,r,t);
}
t=0;
for(int i=1;i<=n;i++){
t+=b[i];//对差分数组求和输出结果
cout<<t<<" ";
}
return 0;
}
代码2 节省一半的存储空间
#include <iostream>
using namespace std;
int b[100005];
int n,m;
//对差分数组进行插入操作
void insert(int l,int r,int v){
b[l]+=v;//a[l~r]的数都相当于加上了一个数v,其他数无影响
b[r+1]-=v;
}
int main()
{
int l,r,t;
cin>>n>>m;
//相当于从全部是0的数组开始构造差分数组
//注意点:不能从0开始存储,题目下标从1,不然会导致错误
for(int i=1;i<=n;i++){
cin>>t;
insert(i,i,t);
}
while(m--){//进行m次差分操作
cin>>l>>r>>t;
insert(l,r,t);
}
t=0;
for(int i=1;i<=n;i++){
t+=b[i];//对差分数组求和输出结果
cout<<t<<" ";
}
return 0;
}