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

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

原始题目链接：https://leetcode.cn/problems/range-sum-query-2d-immutable/

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

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

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

解题思路：

这道题需要先计算二维数组的前缀和，然后利用前缀和来计算任意两个坐标所组成的矩阵面积，具体思路看参考文献，参考文献中的图解真的很赞。

代码实现：

class NumMatrix:

    def __init__(self, matrix: List[List[int]]):
        if not matrix or not matrix[0]:
            row, col = 0, 0
        else:
            row, col = len(matrix), len(matrix[0])

        # 定义prefix_sum[i][j]：从 [0,0]位置（即坐标原点）到[i,j]位置的子矩形所有元素之和
        # 初始化prefix_sum[i][j]
        # prefix_sum 比原矩阵 matrix 多了一行一列，是为了让第 0 行与第 0 列的元素也能使用递推公式
        self.prefix_sum = [[0] * (col + 1) for _ in range(row + 1)]
        # 计算二维数组的前缀和
        for i in range(row):
            for j in range(col):
                self.prefix_sum[i + 1][j + 1] = self.prefix_sum[i][j + 1] + self.prefix_sum[i + 1][j] - self.prefix_sum[i][j] + matrix[i][j]


    
    def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:
        # 计算给定任意两个坐标的矩形区域和
        # 利用prefix_sum进行计算
        return self.prefix_sum[row2 + 1][col2 + 1] - self.prefix_sum[row2 + 1][col1] - self.prefix_sum[row1][col2 + 1] + self.prefix_sum[row1][col1]




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

参考文献：
https://leetcode.cn/problems/range-sum-query-2d-immutable/solution/ru-he-qiu-er-wei-de-qian-zhui-he-yi-ji-y-6c21/