0边界条件/数组中等 LeetCode304. 二维区域和检索 - 矩阵不可变

18阿鲁

已于 2022-04-04 19:53:47 修改

阅读量103

点赞数

文章标签： leetcode

于 2022-04-04 19:43:32 首次发布

本文链接：https://blog.csdn.net/weixin_43260719/article/details/123958781

版权

本文介绍了如何通过预处理二维矩阵实现快速检索子矩阵区域和。通过建立一个额外的dp数组，记录每个位置到原点的元素总和，从而在O(1)时间内计算任意子矩阵的和。在处理边界条件时，确保dp数组比原矩阵多一行和一列以避免越界，简化边界判断。这种优化方法在处理大量查询时能显著提高效率。

摘要由CSDN通过智能技术生成

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

描述

给定一个二维矩阵 matrix，以下类型的多个请求：

计算其子矩形范围内元素的总和，该子矩阵的左上角为 (row1, col1) ，右下角为 (row2, col2) 。
实现 NumMatrix 类：

NumMatrix(int[][] matrix) 给定整数矩阵 matrix 进行初始化
int sumRegion(int row1, int col1, int row2, int col2) 返回左上角 (row1, col1) 、右下角 (row2, col2) 所描述的子矩阵的元素总和。

分析

在sumRegion里面遍历求和随着sumRegion被调用的次数越多，时间复杂度很高，选择求出每个结点到[0,0]这段区域的之和，然后O(1)时间内间接求出目标区域和。
求和时注意不是dp[row2][col2] - dp[row2][col1] - dp[row1][col2] + dp[row1][col1];因为这样会把目标区域的最上面一行和最左面的一列给排出掉了。
应该是dp[row2][col2] - dp[row2][col1-1] - dp[row1-1][col2] + dp[row1-1][col1-1],这样求出的区域才是左上角与右下角围城的区域之和。
dp[row2][col2] - dp[row2][col1-1] - dp[row1-1][col2] + dp[row1-1][col1-1]会有越界的可能，应该在构造dp数组时就比matrix数组多一行多一列，这样可以省去很多边界判断。

class NumMatrix {
    int[][] dp;
    public NumMatrix(int[][] matrix) {
        if(matrix.length == 0){
            return ;
        }
        dp = new int[matrix.length+1][matrix[0].length+1];
        for(int i = 1; i < dp.length; i++){
            for(int j = 1; j < dp[0].length; j++){
               dp[i][j] = dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1] + matrix[i-1][j-1];
            }
        }
    }
    
    public int sumRegion(int row1, int col1, int row2, int col2) {
        return dp[row2+1][col2+1] - dp[row2+1][col1] - dp[row1][col2+1] + dp[row1][col1];
    }
}

/**
 * Your NumMatrix object will be instantiated and called as such:
 * NumMatrix obj = new NumMatrix(matrix);
 * int param_1 = obj.sumRegion(row1,col1,row2,col2);
 */