问题描述
给定一个二维矩阵,计算其子矩形范围内元素的总和,该子矩阵左上角为 (row1, col1) ,右下角为 (row2, col2) 。
上图子矩阵左上角 (row1, col1) = (2, 1) ,右下角(row2, col2) = (4, 3),该子矩形内元素的总和为 8。
示例:
给定 matrix = [
[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]
]
sumRegion(2, 1, 4, 3) -> 8
sumRegion(1, 1, 2, 2) -> 11
sumRegion(1, 2, 2, 4) -> 12
提示:
- 你可以假设矩阵不可变。
- 会多次调用
sumRegion
方法*。* - 你可以假设
row1 ≤ row2
且col1 ≤ col2
。
问题思路
思路1:蛮力法
首先是蛮力法,每次调用sumRange时,通过循环的方法计算数组nums指定区域的元素之和。这种办法运行时间较长,为190ms。
空间复杂度为O(mn),m为数组的行数,n为列数。NumArray的时间复杂度为O(mn),sumRange复杂度为O(mn)。
class NumArray {
int[] sums;
public NumArray(int[] nums) {
sums = nums;
}
public int sumRange(int i, int j) {
int sum = 0;
for(int k = i; k <= j; k++){
sum += sums[k];
}
return sum;
}
}
思路2:动态规划法
和303题相同的思路,由于sumRange会被经常调用,而NumArray则只会调用一次,因此考虑减少sumRange的时间复杂度,在NumArray中对数据进行预处理。
对于数组的每一行,由于sumRange(row, col1, row, col2) = sumRange(row, 0, row, col2) - sumRange(row, 0, row, col1),因此可以考虑计算每个sumRange(row, 0, row, col)的结果,并保存在sums数组中。在调用sumRange时,只需对row1行到row2行用sums数组的元素进行相减后累加即可。
空间复杂度为O(mn),m为数组的行数,n为列数。NumArray的时间复杂度为O(mn),sumRange复杂度为O(m)。运行时间为15ms。
class NumMatrix {
int[][] sums;
public NumMatrix(int[][] matrix) {
int m = matrix.length;
if(m > 0){
int n = matrix[0].length;
sums = new int[m][n+1];
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
sums[i][j+1] = sums[i][j] + matrix[i][j];
}
}
}
}
public int sumRegion(int row1, int col1, int row2, int col2) {
int sum = 0;
for(int i = row1; i <= row2; i++){
sum += sums[i][col2+1] - sums[i][col1];
}
return sum;
}
}
思路3:动态规划法——改进
虽然思路2已经大大降低了sumRange的时间复杂度,但是我们希望能够得到更有结果,将其时间复杂度降为O(1)。
(这里是一个示意图)
由上图可以看到,S = S1 - S2 - S3 + S4 ,且S1,S2,S3,S4都可以通过sumRange(0, 0, row, col)的形式计算。
因此可以先计算出每个sumRange(0, 0, row, col)的结果,保存在sums数组中。在调用sumRange时,由sumRange(row1, col1, row2, col2) = sumRange(0, 0, row2, col2) - sumRange(0, 0, row1-1, col2) - sumRange(0, 0, row2, col1-1) + sumRange(0, 0, row1-1, col1-1) 即可得到结果。
其中,sumRange(0, 0, row, col) = sumRange(0, 0, row-1, col) + sumRange(0, 0, row, col-1) - sumRange(0, 0, row-1, col-1) + matrix(0, 0, row, col)。
空间复杂度为O(mn),m为数组的行数,n为列数。NumArray的时间复杂度为O(mn),sumRange复杂度为O(1)。运行时间为14ms。
class NumMatrix {
int[][] sums;
public NumMatrix(int[][] matrix) {
int m = matrix.length;
if(m > 0){
int n = matrix[0].length;
sums = new int[m+1][n+1];
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
sums[i+1][j+1] = sums[i+1][j] + sums[i][j+1] - sums[i][j] + matrix[i][j];
}
}
}
}
public int sumRegion(int row1, int col1, int row2, int col2) {
return sums[row2+1][col2+1] - sums[row2+1][col1] - sums[row1][col2+1] + sums[row1][col1];
}
}