【9. 子矩阵和】

简介: 思路:>计算子矩阵的和,例如以(x1,y1)为左上角,(x2,y2)为右上角的子矩阵的和为多少。>## 需要解决俩个问题>1. s [ i , j ]如何计算>一行一行的计算>```c>S[i,j] = S[i-1,j] + S[i,j-1] - S[i-1,j-1] + a[i,j]

子矩阵和(二维前缀和)

思路:

计算子矩阵的和,例如以(x1,y1)为左上角,(x2,y2)为右上角的子矩阵的和为多少。

需要解决俩个问题

  1. s [ i , j ]如何计算

一行一行的计算

S[i,j] = S[i-1,j] + S[i,j-1] - S[i-1,j-1] + a[i,j]

1661150526587.png

  1. 以(x1,y1)为左上角,(x2,y2)为右下角这一子矩阵中所有数的和该如何计算
S[x2,y2] -S[x1-1,y2]-S[s2,y1-1]+S[x1-1,y1-1]

1661150549055.png

1661150565187.png

1661150580119.png

题目

输入一个 n 行 mm 列的整数矩阵,再输入 q 个询问,每个询问包含四个整数 x1,y1,x2,y2表示一个子矩阵的左上角坐标和右下角坐标。

对于每个询问输出子矩阵中所有数的和。

输入格式

第一行包含三个整数 n,m,q

接下来 n 行,每行包含 m个整数,表示整数矩阵。

接下来 q行,每行包含四个整数 x1,y1,x2,y2表示一组询问。

输出格式

共 q 行,每行输出一个询问的结果。

数据范围

1 ≤ n ,m ≤ 1000
1 ≤ q ≤ 200000
1≤ x1 ≤ x2 ≤ n
1 ≤ y1 ≤ y2 ≤ m
−1000 ≤ 矩阵内元素的值 ≤ 1000

输入样例:

3 4 3
1 7 2 4
3 6 2 8
2 1 2 3
1 1 2 2
2 1 3 4
1 3 3 4

输出样例:

17
27
21

代码

#include <iostream>
using namespace std;
const int N = 1010;

int n, m, q;
int a[N][N],s[N][N];

int main()
{
   scanf("%d%d%d",&n ,&m, &q);
   for (int i = 1; i <= n; i ++)    //给数组赋值
       for (int j = 1; j <= m; j ++)
           scanf("%d",&a[i][j]);        
           
   //初始化前缀和数组
   for (int i = 1; i <= n; i ++)
       for (int j = 1; j <= m; j ++)
           s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
           
   //输出前缀和
   while (q --)
   {
       int x1,y1,x2,y2;
       scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
       printf("%d\n",s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1]);
   }
   return 0;
}
目录
相关文章
|
5月前
|
资源调度 算法 计算机视觉
BRIEF描述子生成算法
BRIEF描述子生成算法
46 0
|
6月前
|
机器学习/深度学习 存储 人工智能
利用前缀和计算二维矩阵子矩阵的和
利用前缀和计算二维矩阵子矩阵的和
64 0
|
6月前
|
算法 前端开发 测试技术
3033. 修改矩阵
3033. 修改矩阵
36 0
|
6月前
子矩阵的和
子矩阵的和
42 0
|
人工智能 JavaScript 算法
基础算法-子矩阵的和
与前缀和的基本思想类似,上一篇博文对前缀和有所介绍,不是很理解的友友可以阅读一下。一般的前缀和可以看作一维运算,而子矩阵的和则可以看作是二维运算,但本质也是一样的。 原数组可以看作 a[i][j],s[i][j] 是所求区域的数组值和,以下图为例
(二维前缀和模板)796. 子矩阵的和
(二维前缀和模板)796. 子矩阵的和
65 0
|
算法
LeetCode 73矩阵置零&74搜素二维矩阵&75颜色分类
给定一个 m x n 的矩阵,如果一个元素为 0,则将其所在行和列的所有元素都设为 0。请使用原地算法。
129 0
LeetCode 73矩阵置零&74搜素二维矩阵&75颜色分类
|
机器学习/深度学习 算法
二维矩阵转换(多种遍历方法)
二维矩阵转换(多种遍历方法)