【WZOI】（基础题库）第448题 区域修改 2

题目描述:

有一个 N× M 的矩阵 A， 操作 add(x1,y1,x2,y2 , k)表示对矩阵 A 的（x1,y1） 到（x2,y2）区域内的每个 数都加上 k。 有 P 个 add 操作， 输出 P 个 add 操作后的矩阵 A。
 

输入格式:

第一行 3 个正整数： N、 M、 P， 范围[1,1000]。
接下来， N 行 M 列描述矩阵 A，每个数的范围为[1,100]；
后面紧跟 P 行， 每行五个整数， 分别为 x1,y1,x2,y2,k;
1<=x1<=x2<=N， 1<=y1<=y2<=M， k 范围[1,1000]

输出格式:

N 行 M 列的矩阵， 为 add 操作后的新矩阵。

样例输入:

3 3 2
1 2 3
4 5 6
7 8 9
2 1 3 2 4
1 2 2 3 1

样例输出:

1 3 4
8 10 7
11 12 9

提示:

用差分 + 前缀和进行降维， 时间复杂度变为 O(N*M+P)， 1 秒内可轻松出解。

时间限制: 1000ms
空间限制: 256MB

这道题一看就是一道差分题目，还是一道很模板的差分，可以把一个N*M的矩阵看做N个拥有M个元素的一位数组，再在这N个数组中进行差分就行了，所以二维差分就是在一维差分的基础上实现的。

代码实现：

#include<bits/stdc++.h>
using namespace std;
int n,a[1001][1001],b[1001][1001],m,p,x,y,x2,y2,k,ans;
int main(){
	ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
	cin>>n>>m>>p;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++)cin>>a[i][j];
	}
	for(int j=1;j<=p;j++){
		cin>>x>>y>>x2>>y2>>k;
		for(int j=x;j<=x2;j++)b[j][y]+=k,b[j][y2+1]-=k;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m+1;j++){
			b[i][j]+=b[i][j-1];
			a[i][j]+=b[i][j];
		}
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++)cout<<a[i][j]<<" ";
		cout<<"\n";
	}
	return 0;
}