304. 二维区域和检索 - 矩阵不可变（动态规划）

Posted 2021-03-07 cheviszhang

给定一个二维矩阵，计算其子矩形范围内元素的总和，该子矩阵的左上角为 (row1, col1) ，右下角为 (row2, col2)。 来自leetcode

 

 

本题既是一个熟悉class的题目，也是一道简单的动态规划题，子矩阵面积 = 两块面积 - 重复面积 + 右下角面积

收获： 都是一些值得学习的小习惯

　　1. 特判的时候直接return， 我会经常思考要不要 return 0

　　　　def __init__(self,matrix):

　　　　　　if not matrix:

　　　　　　　　return

　　2.动态规划方程，求子面积可以不用每次都算，可以全部算好了再一起取用，以下是学习leetcod ID 吴彦祖的代码

class NumMatrix:

    def __init__(self, matrix: List[List[int]]):

        if(not matrix):

            return

        m=len(matrix)

        n=len(matrix[0])

        self.dp=[[0]*(n+1) for _ in range(m+1)]

        for i in range(1,m+1):

            for j in range(1,n+1):

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

        

    def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:

        return self.dp[row2+1][col2+1]-self.dp[row2+1][col1]-self.dp[row1][col2+1]+self.dp[row1][col1]