题意:
长度为n的数组,操作有q个(1<=n,q<=2*10^5)。每个元素ai范围 [-10^6, 10^6]。每次操作给出l,r,c,表示区间[l,r]各个数加上c(1<=l<=r<=n, -10^5<=c<= 10^5)。输出经过q次操作后的整个数组。
思路:
将原数组a[1:n]变为差分数组b[1:n]:b[1]=a[1],b[i]=a[i]-a[i-1]。这样可以将区间修改转变为单点修改。a[l,r]+=c可以变为b[l]+=c,b[r+1]-=c。最后b数组的前缀和sum{b[1~i]}就是a数组的元素值a[i]。
总结:
一道关于前缀和与差分处理区间修改的题目,数据范围应着重注意,为避免麻烦全开了long long int;同时使用cin要关闭流同步,不然会超时。
代码:
#include <iostream>
using namespace std;
long long int n,q;
long long int a[200001],b[200001]; //题目从1开始
int main()
{
std::ios::sync_with_stdio(false);
cin>>n>>q;
for(long long int i=1;i<=n;i++)
cin>>a[i];
b[1]=a[1];
for(long long int i=2;i<=n;i++)
b[i]=a[i]-a[i-1];
for(long long int i=0;i<q;i++)
{
long long int l,r,c;
cin>>l>>r>>c;
b[l]+=c,b[r+1]-=c;
}
long long int sum=0;
for(long long int i=1;i<=n;i++)
{
sum+=b[i];
cout<<sum<<" ";
}
cout<<endl;
}