題目描述
輸入一個n行m列的整數矩陣,再輸入q個操作,每個操作包含五個整數x1, y1, x2, y2, c,其中(x1, y1)和(x2, y2)表示一個子矩陣的左上角坐標和右下角坐標。
每個操作都要將選中的子矩陣中的每個元素的值加上c。
請你將進行完所有操作后的矩陣輸出
輸入格式
第一行包含整數n,m,q。
接下來n行,每行包含m個整數,表示整數矩陣。
接下來q行,每行包含5個整數x1, y1, x2, y2, c,表示一個操作。
數據范圍
1≤n,m≤1000,
1≤q≤100000,
1≤x1≤x2≤n,
1≤y1≤y2≤m,
?1000≤c≤1000,
?1000≤矩陣內元素的值≤1000
輸入樣例:
3 4 3
1 2 2 1
3 2 2 1
1 1 1 1
1 1 2 2 1
1 3 2 3 2
3 1 3 4 1
輸出樣例:
2 3 4 1
4 3 4 1
2 2 2 2
解題思路
差分初始化,
假定a數組的元素是0
就是對于a數組的每一個元素,[i,j]進行差分
差分的過程
b[x1][y1] += c
b[x2+1][y1] -=c
b[x1][y2+1] -=c
b[x2+1][y2+1] +=c
代碼實現
#include<iostream>
using namespace std;
const int N =1010;
int a[N][N],b[N][N];
int n,m,q;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()
{scanf("%d%d%d",&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;scanf("%d%d%d%d%d",&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)a[i][j] =a[i-1][j] + a[i][j-1] -a[i-1][j-1] +b[i][j];for(int i=1;i<=n;++i){ for(int j=1;j<=m;++j)cout<<a[i][j]<<" ";cout<<endl;} return 0;
}