【中等】力扣算法题解析LeetCode304：二维区域和检索-矩阵不可变

关注文末推广名片，即可免费获得本题测试源码！

题目来源：LeetCode304：二维区域和检索-矩阵不可变

问题抽象： 设计一个类 NumMatrix，用于高效计算二维整数矩阵的任意矩形子区域和（矩阵不可变），满足以下核心需求：

1. 类功能定义：

   • 构造函数：接收二维整数矩阵 matrix 并初始化内部数据结构（矩阵不可变）；
   • 查询方法：sumRegion(int row1, int col1, int row2, int col2) 返回以 (row1, col1) 为左上角、(row2, col2) 为右下角的矩形区域内所有元素之和（包含边界）。

2. 效率约束：

   • 预处理优化：构造函数时间复杂度 O(mn)（m 为行数，n 为列数）；
   • 查询高效：sumRegion 方法时间复杂度 O(1)（避免每次遍历子矩阵）；
   • 空间复杂度 O(mn)：允许额外存储二维数据结构（与输入矩阵同尺寸）。

3. 输入与输出：

   • 输入矩阵 matrix：尺寸 m×n（1 ≤ m, n ≤ 200），元素值 -10^5 ≤ matrix[i][j] ≤ 10^5；
   • 查询参数 row1, col1, row2, col2：满足 0 ≤ row1 ≤ row2 < m，0 ≤ col1 ≤ col2 < n；
   • 输出：矩形区域元素和（整数）。

4. 边界处理：

   • 单元素区域：sumRegion(i,j,i,j) 返回 matrix[i][j]；
   • 全矩阵查询：sumRegion(0,0,m-1,n-1) 返回矩阵总和；
   • 负数和处理：如 [[-1,2],[3,-4]]，sumRegion(0,0,1,1) 返回 0；
   • 大矩阵规模：最大尺寸 200x200=40,000 元素（需高效处理）。

类接口定义：

class NumMatrix {  
    public NumMatrix(int[][] matrix);  // 初始化  
    public int sumRegion(int row1, int col1, int row2, int col2);  // 区域和查询  
}  

应用场景：频繁查询固定矩阵的任意子矩阵和（如图像处理、数据分析系统）。

解题思路

使用 二维前缀和 进行预处理：

1. 初始化前缀和数组 dp：

   • 创建 dp 数组，尺寸为 (m+1) x (n+1)（多一行一列避免边界判断）。
   • dp[i][j] 表示从矩阵左上角 (0,0) 到右下角 (i-1,j-1) 的子矩阵元素总和。
   • 递推公式：

     dp[i][j] = dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1] + matrix[i-1][j-1]
     

2. 查询子矩阵和：

   • 对于查询 (row1, col1, row2, col2)：

     sum = dp[row2+1][col2+1] - dp[row1][col2+1] - dp[row2+1][col1] + dp[row1][col1]
     

   • 原理：总面积减去上方和左侧的矩形，再加回重复减去的左上角矩形。

时间复杂度：初始化：O(m*n)；查询：O(1)（每次）。空间复杂度：O(m*n)

代码实现（Java版）🔥点击下载源码

class NumMatrix {
    private int[][] dp; // 二维前缀和数组

    public NumMatrix(int[][] matrix) {
        if (matrix.length == 0 || matrix[0].length == 0) return;
        int m = matrix.length;
        int n = matrix[0].length;
        dp = new int[m + 1][n + 1]; // 多一行一列避免边界检查
        
        // 构建前缀和数组
        for (int i = 1; i <= m; i++) {
            for (int j = 1; j <= n; 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[row1][col2 + 1] - dp[row2 + 1][col1] + dp[row1][col1];
    }
}

代码说明

1. 初始化：

   • 检查输入矩阵是否为空。
   • 创建 dp 数组（尺寸为 (m+1) x (n+1)），多出的行列用于统一处理边界。
   • 通过动态规划填充 dp 数组：每个位置的值由其上方、左侧和左上角的值推导得出。

2. 查询子矩阵和：

   • 直接通过 dp 数组的四个角点计算子矩阵和：
     • dp[row2+1][col2+1]：整个大矩形（从 (0,0) 到 (row2,col2)）。
     • 减去上方矩形（dp[row1][col2+1]）和左侧矩形（dp[row2+1][col1]）。
     • 加回重复减去的左上角矩形（dp[row1][col1]）。

3. 边界处理：

   • 由于 dp 数组多一行一列，原矩阵位置 (i,j) 对应 dp[i+1][j+1]，查询时需将输入坐标 +1 使用。

提交详情（执行用时、内存消耗）