C++差分
描述
给你一个长度为n的正数数组�1,�2,...��a1,a2,...an.
接下来对这个数组进行m次操作,每个操作包含三个参数l,r,k,代表将数组中��,...��al,...ar部分都加上k。
请输出操作后的数组。
输入描述:
第一行包含两个整数n和m。
第二行包含n个整数表示�1,...��a1,...an
接下来是m行,每行三个整数,分别代表每次操作的参数l,r,k.
1≤�,�≤1051≤n,m≤105
−109≤�[�]≤109−109≤a[i]≤109
1≤�≤�≤�1≤l≤r≤n
−109≤�≤109−109≤k≤109
输出描述:
输出1行,表示m次操作后的�1,...��a1,...an
示例1
输入:
3 2
1 2 3
1 2 4
3 3 -2
输出:
5 6 1
#include<iostream>
using namespace std;
const int N=100010;
long long a[N];//前缀和数组
long long b[N];//差分数组
void insert(int l,int r,int k)//一个相当神奇的函数,既可以用于制作差分数组,又可以加减数字,一个函数可以两用,很精妙
{
b[l]+=k;//在差分数组中改变一个值,前缀和数组中的l后的所有数都会改变
b[r+1]-=k;//使r后的数字不再改变
}
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>a[i];//输入前缀和数组
insert(i,i,a[i]);//使用函数输入差分数组
}
while(m--)//m次的改变
{
int l,r,k;//l,r是改变的范围,k是改变的量
cin>>l>>r>>k;//输入三个数
insert(l,r,k);//改变差分数组
}
for(int i=1;i<=n;i++)
{
b[i]=b[i]+b[i-1];//求前缀和,并把新的前缀和直接存入差分数组中
cout<<b[i]<<' ';//输出处理后的数组
}
}