原题:
输入一个 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,表示一个操作。
输出格式
共 n 行,每行 m 个整数,表示所有操作进行完毕后的最终矩阵。
数据范围
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,差分矩阵为b。则 b 的前缀和就是 a
下图为原矩阵 a:
当要进行(x1,y1) 到(x2,y2)都加 c 的操作,即将图形正中间的蓝色小方块所包含的数据都加 c,要进行以下 4 步:
①先b(x1,y1)+=c ,相当于 图中 a 矩阵(x1,y1)的右下角全部都加 c,即图中的右下角蓝色大矩形;
②绿色和红色区域都多加了 c ,所以要减去。先将绿色区域的数据都减去 c;
③再将红色区域的数据减去 c;
④这时会发现绿色和红色区域重复的小方块的数据减去了两次 c,所以要加上 c;
对应的四句主要算法如下:
- b[x1][y1]+=c;
- b[x2+1][y1]-=c;
- b[x1][y2+1]-=c;
- b[x2+1][y2+1]+=c;
代码:
// 易错!i,j一定要从1开始遍历到n,m,不能从0开始
#include<iostream>
using namespace std;
int b[1010][1010];
int n,m,q;
void solve(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++)
{
int a;
cin>>a;
solve(i,j,i,j,a);
}
// 再在差分矩阵内进行+c操作
while(q--)
{
int x1,y1,x2,y2,c;
cin>>x1>>y1>>x2>>y2>>c;
solve(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]<<' ';
if(j==m) cout<<endl;
}
}