1、题目描述
输入一个n行m列的整数矩阵,再输入q个询问,每个询问包含四个整数x1, y1, x2, y2,表示一个子矩阵的左上角坐标和右下角坐标。
对于每个询问输出子矩阵中所有数的和。
输入格式
第一行包含三个整数n,m,q。
接下来n行,每行包含m个整数,表示整数矩阵。
接下来q行,每行包含四个整数x1, y1, x2, y2,表示一组询问。
输出格式
共q行,每行输出一个询问的结果。
数据范围
1≤n,m≤10001≤n,m≤1000,
1≤q≤2000001≤q≤200000,
1≤x1≤x2≤n1≤x1≤x2≤n,
1≤y1≤y2≤m1≤y1≤y2≤m,
−1000≤矩阵内元素的值≤1000−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
2、分析
二维/多维前缀和
那么这个矩阵的前缀和就是:
应用
求(x2,y2)−(x1,y1)子矩阵的和,类比以上的做法,易得:
3、代码
import java.io.*;
import java.util.*;
public class Main{
static int N = 1010;
static int[][] g;
static int[][] s; //前缀和
public static void main(String[] args) {
Scanner in = new Scanner(new BufferedInputStream(System.in));
int n = in.nextInt(); //行
int m = in.nextInt(); //列
int q = in.nextInt();
g = new int[n+1][m+1];
s = new int[n+1][m+1];
for(int i = 1;i <= n;i ++) {
for(int j = 1;j <= m;j ++)
g[i][j] = in.nextInt();
}
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] + g[i][j]; //矩阵的前缀和
}
while(q -- > 0) {
int x1 = in.nextInt();
int y1 = in.nextInt();
int x2 = in.nextInt();
int y2 = in.nextInt();
int sum = 0; //子矩阵的和
for(int i = x1;i <= x2;i ++) {
for(int j = y1;j <= y2;j ++)
sum = s[x2][y2] - s[x2][y1-1] - s[x1-1][y2] + s[x1-1][y1-1];
}
System.out.println(sum);
}
}
}