给定一个二维矩阵 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]