【Leetcode】1074. Number of Submatrices That Sum to Target

题目地址：

https://leetcode.com/problems/number-of-submatrices-that-sum-to-target/

给定一个$m\times n$矩阵，问其有多少个子矩阵之和是给定数$t$。

对于一个数组求其有多少个子数组和为$t$，这是容易的，可以用前缀和 + HashMap来做。这里只需枚举两行，它们是本轮枚举的矩阵的上下两行，然后将枚举的矩阵压缩成一个数组来看待即可。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    public int numSubmatrixSumTarget(int[][] matrix, int target) {
        int m = matrix.length, n = matrix[0].length;
        int[][] preSum = new int[m + 1][n];
    
        for (int j = 0; j < n; j++) {
            for (int i = 1; i <= m; i++) {
                preSum[i][j] = preSum[i - 1][j] + matrix[i - 1][j];
            }
        }
        
        int res = 0;
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < m; i++) {
            for (int j = i; j < m; j++) {
                int sum = 0;
                map.clear();
                map.put(0, 1);
                for (int k = 0; k < n; k++) {
                    sum += preSum[j + 1][k] - preSum[i][k];
                    res += map.getOrDefault(sum - target, 0);
                    map.put(sum, map.getOrDefault(sum, 0) + 1);
                }
            }
        }
        
        return res;
    }
}

时间复杂度$O(m^{2}n)$，空间$O(mn)$。