码迷,mamicode.com
首页 > 其他好文 > 详细

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

时间:2020-02-24 20:22:29      阅读:67      评论:0      收藏:0      [点我收藏+]

标签:图片   div   简单   etc   一个   范围   检索   计算   技术   

给定一个二维矩阵,计算其子矩形范围内元素的总和,该子矩阵的左上角为 (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]

 

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

标签:图片   div   简单   etc   一个   范围   检索   计算   技术   

原文地址:https://www.cnblogs.com/ChevisZhang/p/12358378.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!