304. 二维区域和检索 - 矩阵不可变(中等)(LCR 013)

Rock在学习

已于 2023-11-11 17:11:08 修改

阅读量32

点赞数

分类专栏：数组文章标签：矩阵算法数据结构 python leetcode

于 2023-11-11 16:17:13 首次发布

本文链接：https://blog.csdn.net/weixin_43606146/article/details/134349919

版权

数组专栏收录该内容

24 篇文章 0 订阅

订阅专栏

文章介绍了在LeetCode问题中如何使用动态规划计算二维矩阵中子矩阵的元素总和，通过预处理前缀和数组来高效解决sumRegion查询问题。实例展示了如何创建NumMatrix类并实现sumRegion方法以获取指定区域的和。

摘要由CSDN通过智能技术生成

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) 所描述的子矩阵的元素总和。

示例 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]

class NumMatrix:
    def __init__(self, matrix: List[List[int]]):
        m, n = len(matrix), len(matrix[0])
        self.presum = [[0] * (n + 1) for _ in range(m + 1)]
        for i in range(1, m + 1):
            for j in range(1, n + 1):
                self.presum[i][j] = self.presum[i - 1][j] + self.presum[i][j - 1] - self.presum[i - 1][j - 1] + matrix[i - 1][j - 1]

    def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:
        return self.presum[row2 + 1][col2 + 1] - self.presum[row2 + 1][col1] - self.presum[row1][col2 + 1] + self.presum[row1][col1]