NumMatrix-二维区域和检索-矩阵不可变

最新推荐文章于 2023-04-16 09:41:21 发布

tangshuai96

最新推荐文章于 2023-04-16 09:41:21 发布

阅读量180

点赞数

分类专栏：算法文章标签：动态规划算法

本文链接：https://blog.csdn.net/tangshuai96/article/details/114278543

版权

算法专栏收录该内容

75 篇文章 1 订阅

订阅专栏

题意

给定一个二维矩阵，计算其子矩形范围内元素的总和，该子矩阵的左上角为 (row1, col1) ，右下角为 (row2, col2)。

在这里插入图片描述

上图子矩阵左上角 (row1, col1) = (2, 1) ，右下角(row2, col2) = (4, 3)，该子矩形内元素的总和为 8。

示例:

给定 matrix = [
[3, 0, 1, 4, 2],
[5, 6, 3, 2, 1],
[1, 2, 0, 1, 5],
[4, 1, 0, 1, 7],
[1, 0, 3, 0, 5]
]

sumRegion(2, 1, 4, 3) -> 8
sumRegion(1, 1, 2, 2) -> 11
sumRegion(1, 2, 2, 4) -> 12

说明:

你可以假设矩阵不可变。
会多次调用 sumRegion 方法。
你可以假设 row1 ≤ row2 且 col1 ≤ col2。

解题思路

在这里插入图片描述

代码演示

class NumMatrix {

    int[][] sums;
    public NumMatrix(int[][] matrix) {
        if(matrix.length==0)
           return;
        else {
            sums=new int[matrix.length][matrix[0].length];
            //第一个矩阵的内容大小就是matrix值
            sums[0][0]=matrix[0][0];
            //子矩阵第一行的以左上角为顶点所有值的和
            for(int i=1;i<matrix[0].length;i++)
                sums[0][i]=sums[0][i-1]+matrix[0][i];
            //同上，第一行改为第一列
            for(int i=1;i<matrix.length;i++)
                sums[i][0]=sums[i-1][0]+matrix[i][0];
            //开始计算矩阵的其他部分，以子矩阵的左上角为起点
            for(int i=1;i<matrix.length;i++)
                for(int j=1;j<matrix[0].length;j++)
                {
                    sums[i][j]=sums[i-1][j]+sums[i][j-1]-sums[i-1][j-1]+matrix[i][j];
                }
        }

    }
    
    public int sumRegion(int row1, int col1, int row2, int col2) {
        //如果从子矩阵的左上角开始计算
          if(row1==0&&col1==0)
              return sums[row2][col2];
          //起始点的横坐标是第一行
          if(row1==0)
              return sums[row2][col2]-sums[row2][col1-1];
          //起始点的纵坐标在第一列
          if(col1==0)
              return sums[row2][col2]-sums[row1-1][col2];
          //一般情况
          return sums[row2][col2]-sums[row1-1][col2]-sums[row2][col1-1]+sums[row1-1][col1-1];
    }
}