java---矩阵/二维前缀和算法(每日一道算法2022.7.22)

注意事项:
本题为"一维前缀和"的扩展题,建议先阅读那篇文章并理解。

题目
输入一个 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/
本文仅用作记录和交流

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值