输入一个长度为 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
https://www.acwing.com/problem/content/799/
#include<bits/stdc++.h>
using namespace std;
int main() {
int n,m;
cin>>n>>m;
int a[n+1],s[n+1];
s[0]=0;
for(int i=1;i<=n;i++) {
cin>>s[i];
a[i]=s[i]-s[i-1];
}
int l,r,c;
while(m--) {
cin>>l>>r>>c;
a[l]+=c;
a[r+1]-=c;
}
for(int i=1;i<=n;i++) {
s[i]=s[i-1]+a[i];
cout<<s[i]<<" ";
}
cout<<"\n";
return 0;
}
新的理解:
一开始的插入初始值可以看作是对原数组的操作,即对 [ 1 , 1 ] , [ 2 , 2 ] , [ 3 , 3 ] , [ 4 , 4 ] , [ 5 , 5 ] , … 的更新
#include<bits/stdc++.h>
using namespace std;
int main() {
int n,m;
cin>>n>>m;
int a[n+1];
memset(a,0,sizeof(a));
int t;
for(int i=1;i<=n;i++) {
cin>>t;
a[i]+=t;
a[i+1]-=t;
}
int l,r,c;
while(m--) {
cin>>l>>r>>c;
a[l]+=c;
a[r+1]-=c;
}
int s[n+1];
s[0]=0;
for(int i=1;i<=n;i++) {
s[i]=s[i-1]+a[i];
cout<<s[i]<<" ";
}
cout<<"\n";
return 0;
}
那么显然,前缀数组就不需要了
#include<bits/stdc++.h>
using namespace std;
int main() {
int n,m;
cin>>n>>m;
int a[n+1];
memset(a,0,sizeof(a));
int t;
for(int i=1;i<=n;i++) {
cin>>t;
a[i]+=t;
a[i+1]-=t;
}
int l,r,c;
while(m--) {
cin>>l>>r>>c;
a[l]+=c;
a[r+1]-=c;
}
int sb=0,sf=0;
for(int i=1;i<=n;i++) {
sf=sb+a[i];
cout<<sf<<" ";
sb=sf;
}
cout<<"\n";
return 0;
}
再直接对原数组操作
#include<bits/stdc++.h>
using namespace std;
int main() {
int n,m;
cin>>n>>m;
int a[n+1];
memset(a,0,sizeof(a));
int t;
for(int i=1;i<=n;i++) {
cin>>t;
a[i]+=t;
a[i+1]-=t;
}
int l,r,c;
while(m--) {
cin>>l>>r>>c;
a[l]+=c;
a[r+1]-=c;
}
for(int i=1;i<=n;i++) {
a[i]+=a[i-1];
cout<<a[i]<<" ";
}
cout<<"\n";
return 0;
}