二维差分矩阵

798. 差分矩阵 - AcWing题库

AC代码：
 

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 1010;
int n,m,q;
int a[N][N],b[N][N];

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()
{
    cin>>n>>m>>q;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;j++)
        {
            cin>>a[i][j];
        }
    }
    
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=m;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];
            cout<<b[i][j]<<" ";
        }
        cout<<endl;
    }
     
     
    return 0;
}

相关解释：
这里构造差分矩阵其实就是在全部都是0的b矩阵中插入a[i][j]，这样就可以构造出差分矩阵，注意insert函数的写法，这样在差分矩阵中，可以将左上角为x1,y1,右下角x2,y2的矩阵中每个元素加个c。这里构造差分矩阵就把把每个点想象成一个矩阵，比如只插入一个a[2][3]时，将刚开始全0的b数组，每个元素的加或减，这样求b数组的前缀和的时候，只有a[2][3]为数，其余都是0.以此类推，可以a数组的每个元素插入到b数组中，这样b数组的前缀和就是a数组，可以得到整个差分矩阵。然后依次q个insert，最后输出一下差分矩阵的前缀和即为最后的a数组。