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
差分
#include<bits/stdc++.h> // 包含大部分标准库
using namespace std;
const int z = 1010; // 设置数组的最大长度,因为数据范围是1≤n,m≤1000
int a[z][z], b[z][z]; // a是原始的矩阵,b是差分矩阵
int main()
{
int n, m, q; // n是行数,m是列数,q是查询次数
scanf("%d %d %d", &n, &m, &q); // 读入n, m, q
int i, j;
for (i = 1; i <= n; i++) // 从1开始,因为差分矩阵需要从(1,1)开始计算
{
for (j = 1; j <= m; j++)
{
scanf("%d", &a[i][j]); // 读入矩阵元素值
// 计算差分矩阵,当前元素等于原始矩阵中当前元素减去上方元素减去左侧元素再加上左上角元素
//s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+a[i][j];
b[i][j] = a[i][j] - a[i - 1][j] - a[i][j - 1] + a[i - 1][j - 1];
}
}
while (q--) // 循环处理每个操作
{
int x1, y1, x2, y2, c; // 操作的子矩阵的左上角和右下角坐标,以及要加上的值c
scanf("%d %d %d %d %d", &x1, &y1, &x2, &y2, &c); // 读入操作
b[x1][y1] += c; // 在左上角加上c
b[x1][y2 + 1] -= c; // 将右上角的下一个元素减去c
b[x2 + 1][y1] -= c; // 将左下角的下一个元素减去c
b[x2 + 1][y2 + 1] += c; // 将右下角的下一个元素加上c
}
for (i = 1; i <= n; i++)
{
for (j = 1; j <= m; j++)
{
a[i][j] = a[i - 1][j] + a[i][j - 1] - a[i - 1][j - 1] + b[i][j]; // 计算最终矩阵
printf("%d ", a[i][j]); // 输出最终矩阵的每个元素
}
printf("\n"); // 输出换行符
}
return 0;
}
这段代码的核心思想是差分矩阵。差分矩阵是一种特殊的矩阵,它记录了原始矩阵每个元素与其左上角元素之差的变化。通过对差分矩阵进行一系列操作,可以快速更新原始矩阵中的元素。
在代码中,首先读入原始矩阵,并计算出差分矩阵。差分矩阵的计算是通过当前元素等于原始矩阵中当前元素减去上方元素减去左侧元素再加上左上角元素得到的。
接下来,根据输入的操作,对差分矩阵进行更新。每个操作表示一个子矩阵,将子矩阵中的每个元素都加上给定的值c。这是通过将子矩阵的四个角进行相应的操作得到的:左上角加上c,右上角的下一个元素减去c,左下角的下一个元素减去c,右下角的下一个元素加上c。
最后,通过差分矩阵和原始矩阵的关系,计算出最终的矩阵,并输出每个元素。最终矩阵的计算是通过原始矩阵中每个元素等于上方元素加上左侧元素减去左上角元素再加上差分矩阵中对应位置的元素得到的。