可参考
输入一个长度为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
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int n, m;
int a[N], b[N];
void insert(int start, int end, int number)
{
b[start] += number;
b[end + 1] -= number;
}
int main()
{
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; ++i) {
scanf("%d", &a[i]);
}
for (int i = 1; i <= n; ++i) {
insert(i, i, a[i]);
}
while (m--) {
int l = 0, r = 0, c = 0;
scanf("%d%d%d", &l, &r, &c);
insert(l, r, c);
}
for (int i = 1; i <= n; ++i) {
b[i] += b[i - 1];
}
for (int i = 1; i <= n; ++i) {
printf("%d ", b[i]);
}
printf("\n");
return 0;
}
倘若已经升级到了二维数组
创建一个新数组b,b[i][j]=sum(a[i][j+1]+…a[maxn][maxn])
那么核心的代码就是
对差分数组的操作:
b[x1][y1]+=c
b[x1][y2+1]-=c
b[x2+1][y1]-=c
b[x2+1][y2+1]+=c
关于这个的解释可见博客
#include<iostream>
using namespace std;
const int maxn = 1e3+10;
int a[maxn][maxn],b[maxn][maxn];
void insert(int x1,int y1,int x2,int y2,int c)
{
b[x1][y1] += c;
b[x2+1][y1] -= c;
b[x1][y2+1] -= c;
b[x2+1][y2+1] += c;
}
int main(){
int n,m,q;
for(int i = 1;i <= n;++i)
for(int j = 1;j <= m;++j)
{
cin >> a[i][j];
insert(i,j,i,j,a[i][j]);
}
while(q--)
{
int x1,y1,x2,y2,c;
cin >> x1 >> y1 >> x2 >> y2 >> c;
insert(x1,y1,x2,y2,c);
}
for(int i = 1;i <= n;++i) {
for(int j = 1;j <= m;++j){
b[i][j] += b[i-1][j]+b[i][j-1]-b[i-1][j-1];
if(j == m) cout << b[i][j];
else cout << b[i][j] << " ";
}
cout << endl;
}
return 0;
}