题目描述:
题解:
矩阵前缀和
1.求坐标(0,0)到(i,j)的matrix对应位置之和。
图中红色部分=两个绿色部分相加-蓝色部分+[i,j]
用thesum[i][j]表示坐标[i,j]到[0,0]的和,thessum[i][j] = thesum[i-1][j]+thesum[i][j-1]-thesum[i-1][j-1]+,matrix[i][j]
2.求坐标[i,j]到[m,n]值之和。
[m][n]到[i][j]矩阵值之和=红色部分-两个绿色部分+蓝色部分
=thesum[i][j]=thesum[i][n]-thesum[m][j]+thesum[m][n]
class NumMatrix:
def __init__(self, matrix: List[List[int]]):
if not matrix or not matrix[0]:
m, n = 0, 0
else:
m = len(matrix)
n = len(matrix[0])
self.thesum = [[0 for i in range(n+1)]for j in range(m+1)]
for i in range(m):
for j in range(n):
self.thesum[i+1][j+1] = self.thesum[i][j+1]+self.thesum[i+1][j]-self.thesum[i][j]+matrix[i][j]
def sumRegion(self, row1: int, col1: int, row2: int, col2: int) -> int:
return self.thesum[row2+1][col2+1]-self.thesum[row2+1][col1]-self.thesum[row1][col2+1]+self.thesum[row1][col1]



