差分矩阵
作者:林小鹿
链接:https://www.acwing.com/solution/content/27325/
来源:AcWing
本人所写代码的差分矩阵数组为s
C++ 代码
#include<iostream>
using namespace std;
const int N=1010;
int a[N][N],s[N][N];
int main()
{
int n,m,q;
scanf("%d%d%d",&n,&m,&q);
for(int i=1;i<=n;i++)//构造差分矩阵,使a[i][j]为s[i][j]的前缀和
{
for(int j=1;j<=m;j++)
{
scanf("%d",&a[i][j]);
s[i][j]=a[i][j]-a[i-1][j]-a[i][j-1]+a[i-1][j-1];//即a[i][j]=s[0:i][0:j]
}
}
while(q--)
{
int x1,x2,y1,y2,c;
scanf("%d%d%d%d%d",&x1,&y1,&x2,&y2,&c);
s[x1][y1] += c;
s[x1][y2 + 1] -= c;
s[x2 + 1][y1] -= c;
s[x2 + 1][y2 + 1] += c;
}
for(int i=1;i<=n;i++)//将s数组变为s数组的前缀和,因为s[1][1]为本身的前缀和,所以自下而上,从左到右运算即可求出前缀和
{
for(int j=1;j<=m;j++)
{
s[i][j]+=s[i-1][j]+s[i][j-1]-s[i-1][j-1];
}
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
printf("%d ",s[i][j]);
}
printf("\n");
}
return 0;
}