LeetCode—304. 二维区域和检索 - 矩阵不可变

304. 二维区域和检索 - 矩阵不可变

题目描述：
给定一个二维矩阵 matrix，以下类型的多个请求：

计算其子矩形范围内元素的总和，该子矩阵的 左上角 为 (row1, col1) ，右下角 为 (row2, col2) 。
实现 NumMatrix 类：

NumMatrix(int[][] matrix) 给定整数矩阵 matrix 进行初始化
int sumRegion(int row1, int col1, int row2, int col2) 返回 左上角 (row1, col1) 、右下角 (row2, col2) 所描述的子矩阵的元素 总和 。

考察重点：(前缀和、区域和检索) 先将0-i(0<=i<=n)的和都存储于数组sum中，即到第i位之和为sum[i]；i 到 j 之和==sum[j]-sum[i-1]，二维数组同理。

func Constructor9(matrix [][]int) NumMatrix { //sumMatrix[i][j]保存从(0,0)到(i,j)的元素和
	for i := 0; i < len(matrix); i++ {
		if i > 0 { //(0,0) = (0,0)
			matrix[i][0] += matrix[i-1][len(matrix[0])-1] //换行之后，(i,j)等于(i-1,len)+(i,j)
		}
		for j := 1; j < len(matrix[0]); j++ { //其他位置，(i,j)等于(i-1,j)+(i,j)
			matrix[i][j] += matrix[i][j-1]
		}
	}
	return NumMatrix{sumMatrix: matrix}
}

func (t *NumMatrix) SumRegion(row1 int, col1 int, row2 int, col2 int) int {
	sum := 0
	for i := row1; i <= row2; i++ {
		//考虑左上角(0,0)到右下角(i,j)的情况 以及 左上角(i,0)到右下角(i,j)
		if col1 == 0 {
			if i == 0 {
				sum += t.sumMatrix[i][col2] //原点直接加
				continue
			} //(i,j)等于 (i,j)-(i-1,len)
			sum += (t.sumMatrix[i][col2] - t.sumMatrix[i-1][len(t.sumMatrix[0])-1])
			continue
		}
		sum += (t.sumMatrix[i][col2] - t.sumMatrix[i][col1-1])
	}
	return sum
}