AcWing--798.差分矩阵(差分)

原题:

输入一个 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;

对应的四句主要算法如下:

  1. b[x1][y1]+=c;
  2. b[x2+1][y1]-=c;
  3. b[x1][y2+1]-=c;
  4. 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;
			}
}
  • 8
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值