题目描述
小明拥有 N个彩灯,第 i 个彩灯的初始亮度为 ai。 小明将进行 Q 次操作,每次操作可选择一段区间,并使区间内彩灯的亮度 +x ( 可能为负数)。 求 Q 次操作后每个彩灯的亮度(若彩灯亮度为负数则输出 0 )。
输入描述
第一行包含两个正整数 N,Q ,分别表示彩灯的数量和操作的次数。
第二行包含 N个整数,表示彩灯的初始亮度。
接下来 Q 行每行包含一个操作,格式如下:
l r x ,表示将区间 l~r 的彩灯的亮度 +x 。
输出描述 输出共 1 行,包含 N 个整数,表示每个彩灯的亮度。
题解代码
定义 b1=a1;bi=ai-a(i-1) i>=2 那么 b 就是 a 的差分数组,假设现在 al-ar
同时加上 x ,那么 b 数组的变化是什么样的呢 ?
简单的模拟一下发现 b(r+1) 加上了 -x , bl 加上了 x , 数组就只修改了俩个数,那么我在 b 数组上修改 怎么回到 a 数组呢?根据 b 数组的定义,我们发现其实 a 就是 b 的前缀和数组,也就是在 b 上修改完对 b 进 行前在 b上修改完对 进行前缀和就可以得到修改完的 a 数组
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=5e5+10;
ll a[N],ans[N]; //a[N]为差分数组
int main()
{
ll n,q;//n个彩灯,q次操作
cin>>n>>q;
for(ll i=1;i<=n;i++) cin>>ans[i];
while(q--)
{
ll l,r,x;
cin>>l>>r>>x;
a[l]+=x,a[r+1]-=x;
}
for(ll i=1;i<=n;i++) a[i]+=a[i-1];
for(ll i=1;i<=n;i++)
cout<<max(a[i]+ans[i],0ll)<<" ";
return 0;
}