前缀和与差分 AcWing 798. 差分矩阵
前缀和与差分 AcWing 798. 差分矩阵
原题链接
AcWing 798. 差分矩阵
算法标签
差分
思路
改图转该题解
代码
#include<bits/stdc++.h>
#define int long long
#define rep(i, a, b) for(int i=a;i<b;++i)
#define Rep(i, a, b) for(int i=a;i>b;--i)
using namespace std;
const int N = 1005;
int a[N][N],b[N][N];
inline int read(){
int s=0,w=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
while(ch>='0'&&ch<='9') s=s*10+ch-'0',ch=getchar();
return s*w;
}
void put(int x) {
if(x<0) putchar('-'),x=-x;
if(x>=10) put(x/10);
putchar(x%10^48);
}
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;
}
signed main(){
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
int n=read(), m=read(), q=read();
rep(i, 1, n+1){
rep(j, 1, m+1){
a[i][j]=read();
// b数组初始化
insert(i, j, i, j, a[i][j]);
}
}
// 差分
while(q--){
int a=read(), b=read(), c=read(), d=read(), e=read();
insert(a, b, c, d, e);
}
// 最终矩阵
rep(i, 1, n+1){
rep(j, 1, m+1){
b[i][j]+=b[i-1][j]+b[i][j-1]-b[i-1][j-1];
}
}
rep(i, 1, n+1){
rep(j, 1, m+1){
printf("%lld ", b[i][j]);
}
puts("");
}
return 0;
}
模板
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;
}
原创不易
转载请标明出处
如果对你有所帮助 别忘啦点赞支持哈
还没有评论,来说两句吧...