给出矩阵 matrix 和目标值 target,返回元素总和等于目标值的非空子矩阵的数量。
子矩阵 x1, y1, x2, y2 是满足 x1 <= x <= x2 且 y1 <= y <= y2 的所有单元 matrix[x][y] 的集合。
如果 (x1, y1, x2, y2) 和 (x1', y1', x2', y2') 两个子矩阵中部分坐标不同(如:x1 != x1'),那么这两个子矩阵也不同。
示例 1:
输入:matrix = [[0,1,0],[1,1,1],[0,1,0]], target = 0
输出:4
解释:四个只含 0 的 1x1 子矩阵。
示例 2:
输入:matrix = [[1,-1],[-1,1]], target = 0
输出:5
解释:两个 1x2 子矩阵,加上两个 2x1 子矩阵,再加上一个 2x2 子矩阵。
提示:
1 <= matrix.length <= 300
1 <= matrix[0].length <= 300
-1000 <= matrix[i] <= 1000
-10^8 <= target <= 10^8
思路:以行为单位分别求出每行的前缀和(以列为单位当然也可以),之后通过枚举列的起点和重点来暴力子矩阵,为了避免复杂度达到m^2*n^2,我们考虑用map优化一波。
class Solution {
public int numSubmatrixSumTarget(int[][] matrix, int target) {
int ans = 0;
int m = matrix.length;
int n = matrix[0].length;
int[][] sum = new int[m + 1][n + 1];
Map<Integer, Integer> count = new HashMap<>();
for (int i = 1; i <= m; i++)
for (int j = 1; j <= n; j++)
sum[i][j] = sum[i][j - 1] + matrix[i - 1][j - 1];
int endY = 1;
while (endY <= n) {
for (int startY = 0; startY < endY; startY++) {
int now = 0;
count.clear();
count.put(0, 1);
for (int i = 1; i <= m; i++) {
now += sum[i][endY] - sum[i][startY];
if (count.containsKey(now - target))
ans += count.get(now - target);
count.put(now, count.getOrDefault(now, 0) + 1);
}
}
endY++;
}
return ans;
}
}