剑指 Offer II 013. 二维子矩阵的和

概述

看我这图，一眼就清楚了。

题目

链接：https://leetcode.cn/problems/O4NDxx

给定一个二维矩阵 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[][] matrix;

public NumMatrix(int[][] matrix) {
    this.matrix = matrix;
}

public int sumRegion(int row1, int col1, int row2, int col2) {
    int result = 0;
    for (int i = row1; i <= row2; i++) {
        int[] matrix = this.matrix[i];
        for (int j = col1; j <= col2; j++) {
            result += matrix[j];
        }
    }
    return result;
}

解法二：预处理矩阵

逻辑

我们进一步优化一下，想一想，是不是可以按照前几题那样，通过加和以及加和之间的差，直接计算子矩阵的和呢？

看图找规律：

如果我们想求蓝色子矩阵的和，那么用整个矩阵的和-红框子矩阵的和-黄框子矩阵的和+绿色子矩阵的和，就可以了。

所以我们预处理矩阵时，只需要先遍历计算出(0,0)到每个元素之间形成的子矩阵的面积和，就可以在最终计算时派上用场了。

注意，这里在处理的时候，是按照逐行去处理。

代码

int[][] sumMatrix;
public NumMatrix(int[][] matrix) {
  int[][] sumMatrix = new int[matrix.length + 1][matrix[0].length + 1];
  for (int i = 0; i < matrix.length; i++) {
    int rowSum = 0;
    for (int j = 0; j < matrix[0].length; j++) {
      rowSum += matrix[i][j];
      sumMatrix[i + 1][j + 1] = sumMatrix[i][j + 1] + rowSum;
    }
  }

  this.sumMatrix = sumMatrix;
}

public int sumRegion(int row1, int col1, int row2, int col2) {
  return sumMatrix[row2 + 1][col2 + 1] - sumMatrix[row1][col2 + 1] - sumMatrix[row2 + 1][col1] + sumMatrix[row1][col1];
}