力扣 304.二维区域和检索-数组不可变

问题描述

给定一个二维矩阵，计算其子矩形范围内元素的总和，该子矩阵左上角为 (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];
    }
}