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

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 。

---

解决方案：

• 思路一：
  获取匹配>=row1 && <= row2行，累加每行 >= col1 && <=col2 的值

/**
 * @param {number[][]} matrix
 */
var NumMatrix = function(matrix) {
   this.arr = matrix;
};

/** 
 * @param {number} row1 
 * @param {number} col1 
 * @param {number} row2 
 * @param {number} col2
 * @return {number}
 */
NumMatrix.prototype.sumRegion = function(row1, col1, row2, col2) {
     let num = 0;
     for (let i = 0; i < this.row; i++) {
         if (i >= row1 && i <= row2) {
             for (let j = 0; j < this.col; j++) {
                 let da = 0;
                 if (j >= col1 && j <= col2) {
                          da += this.arr[i][j];
                     }
                     num += da
                 }
             }
     }
     return num;
};

结果

• 思路二：
  转换一个每格行列相加的矩阵，目标小矩形和 = 右下角最大面积和 - 左下角左边面积和 - 右上角上边面积和 + 上左上角被重复减去的面积

/**
* @param {number[][]} matrix
*/
var NumMatrix = function(matrix) {
  const rows = matrix.length
   if (!rows) {
       return matrix
   }
   const columns = matrix[0].length
   this.sums = JSON.parse(JSON.stringify(new Array(rows + 1).fill(new Array(columns + 1).fill(0))))
   for (var i = 1; i < rows + 1; i++) {
       for (var j = 1; j < columns + 1; j++) {
           this.sums[i][j] = this.sums[i - 1][j] + this.sums[i][j - 1] - this.sums[i - 1][j - 1] + matrix[i - 1][j - 1]
       }
   }
};

/** 
* @param {number} row1 
* @param {number} col1 
* @param {number} row2 
* @param {number} col2
* @return {number}
*/
NumMatrix.prototype.sumRegion = function(row1, col1, row2, col2) {
   return this.sums[row2 + 1][col2 + 1] - this.sums[row2 + 1][col1] - this.sums[row1][col2 + 1] + this.sums[row1][col1]
};