将二维前缀和与差分组合在一起的一个比较综合的算法
描述
给你一个n行m列的矩阵,下标从1开始。
接下来有q次操作,每次操作输入5个参数x1, y1, x2, y2, k
表示把以(x1, y1)为左上角,(x2,y2)为右下角的子矩阵的每个元素都加上k,
请输出操作后的矩阵。
输入描述:
第一行包含三个整数n,m,q.
接下来n行,每行m个整数,代表矩阵的元素
接下来q行,每行5个整数x1, y1, x2, y2, k,分别代表这次操作的参数
1≤�,�≤10001≤n,m≤1000
1≤�≤1051≤q≤105
1≤�1≤�2≤�1≤x1≤x2≤n
1≤�1≤�2≤�1≤y1≤y2≤m
−109≤矩阵中的元素≤109−109≤矩阵中的元素≤109
输出描述:
输出n行,每行m个数,每个数用空格分开,表示这个矩阵。
示例1
输入:
2 3 4
1 2 3
4 5 6
1 1 2 2 3
1 2 2 3 -1
1 1 1 3 4
1 1 2 1 1
输出:
9 8 6
8 7 5
#include<iostream>
using namespace std;
const int N=1010;
long long a[N][N];//前缀和数组
long long b[N][N];//差分数组
void insert(int x1,int y1,int x2,int y2,int k)
{
b[x1][y1]+=k;//右下侧全部加k
b[x2+1][y1]-=k;//下侧全部减去k
b[x1][y2+1]-=k;//右侧全部减去k
b[x2+1][y2+1]+=k;//最右下侧小矩形全部加上k
}
int main()
{
int n,m,q;
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--)//循环q次
{
int x1,y1,x2,y2,k;//输入5个变量
cin>>x1>>y1>>x2>>y2>>k;
insert(x1,y1,x2,y2,k);//改变差分数组
}
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];//将改变后差分数组的前缀和存入差分数组中
//输出
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
cout<<b[i][j]<<' ';
cout<<endl;
}
}