leetcode 剑指 Offer II 013. 二维子矩阵的和

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

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

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

示例 1：

输入:
[“NumMatrix”,“sumRegion”,“sumRegion”,“sumRegion”]
[[[[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]]],[2,1,4,3],[1,1,2,2],[1,2,2,4]]
输出:
[null, 8, 11, 12]

解释:
NumMatrix numMatrix = new NumMatrix([[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]]]);
numMatrix.sumRegion(2, 1, 4, 3); // return 8 (红色矩形框的元素总和)
numMatrix.sumRegion(1, 1, 2, 2); // return 11 (绿色矩形框的元素总和)
numMatrix.sumRegion(1, 2, 2, 4); // return 12 (蓝色矩形框的元素总和)

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/O4NDxx

思路

• 大红色区域子矩阵的和matrix[i1,j1,i2,j2] 与四个部分相关
• dp[i,j] 表示 matrix[0,0,i,j] 中二维矩阵的和
• 将第一行和第一列进行增量累加，初始化dp数组
• 初始化dp的其他位置 dp[i,j]，和dp[i-1][j],dp[i][j-1],dp[i-1][j-1],matrix[i][j] 相关
  

题解

class NumMatrix:

    def __init__(self, matrix):
        self.matrix = matrix
        self.mt = matrix
        for j in range(1, len(self.mt[0])):
            self.mt[0][j] += self.mt[0][j - 1]
        for j in range(1, len(self.mt)):
            self.mt[j][0] += self.mt[j - 1][0]

        for i in range(1, len(self.mt)):
            for j in range(1, len(self.mt[0])):
                self.mt[i][j] += self.mt[i - 1][j] + self.mt[i][j - 1] - self.mt[i - 1][j - 1]

    def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:
        if row1 == 0:
            return self.mt[row2][col2] - self.mt[row2][col1-1] if col1!=0 else self.mt[row2][col2]
        if col1 ==0:
            return self.mt[row2][col2] - self.mt[row1 - 1][col2]
        return self.mt[row2][col2] - self.mt[row1-1][col2] - self.mt[row2][col1-1] + self.mt[row1-1][col1-1]