304 二维区域和检索 - 矩阵不可变（二维前缀和）

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 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-sum-query-2d-immutable/

2. 思路分析：

分析题目可以知道这道题目属于二维前缀和的裸题，与一维前缀和类似，我们也需要一个辅助的二维数组s存储当前的（x， y）到（0，0）对应的矩阵区域和，这样当需要计算（x1，y1）和（x2，y2）构成的矩阵的区域和的时候那么可以直接在s中计算出结果返回即可，主要是加上相应的矩阵区域和并且减去重复计算的矩阵区域和，画一个简单的矩阵会很好理解。

 （画图其实可以很快写出来相关的公式）

3. 代码如下：

from typing import List


class NumMatrix:
    def __init__(self, matrix: List[List[int]]):
        r, c = len(matrix), len(matrix[0])
        # s为前缀和列表
        s = [[0] * (c + 1) for i in range(r + 1)]
        for i in range(1, r + 1):
            for j in range(1, c + 1):
                # matrix下标从0开始
                s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + matrix[i - 1][j - 1]
        # 初始化列表
        self.s = s

    def sumRegion(self, x1: int, y1: int, x2: int, y2: int) -> int:
        # 计算二维列表的前缀和
        # 下标从1开始
        x1 += 1
        y1 += 1
        x2 += 1
        y2 += 1
        s = self.s
        return s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1]