注意事项:
本题为"一维前缀和"的扩展题,建议先阅读那篇文章并理解。
题目
输入一个 n 行 m 列的整数矩阵,再输入 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
测试输入,分别为(x1,y1),(x2,y2)的坐标
1 1 2 2
2 1 3 4
1 3 3 4
输出
17
27
21
public class 二维前缀和 {
public static void main(String[] args) throws IOException {
//用bufferedreader接受数据,使用readline接受到Stirng对象,再用split进行分割
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
String[] init = in.readLine().split(" ");
int n = Integer.parseInt(init[0]), m = Integer.parseInt(init[1]), times = Integer.parseInt(init[2]);
// 传入矩阵数据
int[][] a = new int[n+1][m+1];
for (int i = 1; i<=n; i++) {
String[] matrix = in.readLine().split(" ");
for (int j = 1; j<=m; j++) {
a[i][j] = Integer.parseInt(matrix[j-1]);
}
}
// 计算矩阵前缀和:x2y1 + x1y2 - x1y1 + x2y2
int[][] s = new int[n+1][m+1];
for (int i = 1; i <= n; i++) {
for (int j = 1; j <= m; j++) {
s[i][j] = s[i][j-1]+s[i-1][j]-s[i-1][j-1]+a[i][j];
}
}
//计算给出区域内的矩阵和: x2y2 - x1-1y2 - x2y1-1 + x1-1y1-1
while (times-- > 0) {
String[] data = in.readLine().split(" ");
int x1 = Integer.parseInt(data[0]), y1 = Integer.parseInt(data[1]);
int x2 = Integer.parseInt(data[2]), y2 = Integer.parseInt(data[3]);
System.out.println(s[x2][y2]-s[x2][y1-1]-s[x1-1][y2]+s[x1-1][y1-1]);
}
}
}
思路:
思想上和一维的前缀和其实差不多,只是二维稍显复杂,核心思想是一样的
从大佬那里拿的一张特别好的解释矩阵和的图:
声明:
算法思路来源为y总,详细请见https://www.acwing.com/
本文仅用作记录和交流