给定一个二维矩阵 matrix,以下类型的多个请求:
计算其子矩形范围内元素的总和,该子矩阵的 左上角 为 (row1, col1) ,右下角 为 (row2, col2) 。
实现 NumMatrix 类:
NumMatrix(int[][] matrix) 给定整数矩阵 matrix 进行初始化
int sumRegion(int row1, int col1, int row2, int col2) 返回 左上角 (row1, col1) 、右下角 (row2, col2) 所描述的子矩阵的元素 总和 。
//注意将数组整成int[n+1][m+1] 这样的话可以更好的计算[0][0],[0][0]的值
class NumMatrix {
int[][] sum;
public NumMatrix(int[][] matrix) {
int n = matrix.length;
int m = matrix[0].length;
if (n>0 && m>0){
sum = new int[n+1][m+1];
//二维前缀和中存储的每个值实际上是从最左上角一直到右下角所有值的和
for (int i=1;i<=matrix.length;i++){
for (int j=1;j<=matrix[0].length;j++){
sum[i][j] = matrix[i-1][j-1]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
}
}
}
}
public int sumRegion(int row1, int col1, int row2, int col2) {
return sum[row2+1][col2+1]+sum[row1][col1]-sum[row1][col2+1]-sum[row2+1][col1];
}
}