差分(一维与二维)

一维

原题链接

#include <bits/stdc++.h>
using namespace std;

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

void insert(int l,int r,int c){
    b[l]+=c;
    b[r+1]-=c;
}

int main(){
    cin>>n>>m;
    for(int i=0;i<n;i++){
        cin>>a[i];
        insert(i,i,a[i]);//b[i]==a[i]-a[i-1] 
    }
    
    while(m--){
        int l,r,c;
        cin>>l>>r>>c;
        insert(l-1,r-1,c);
    }
    
    for(int i=0;i<n;i++){
        b[i]+=b[i-1];//a[i]==b[i]+a[i-1]
        cout<<b[i]<<" ";
    }
    
    return 0;
}

二维

原题链接

#include <bits/stdc++.h>
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];
            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;
    
}