元素和为目标值的子矩阵数量

元素和为目标值的子矩阵数量

给出矩阵 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’），那么这两个子矩阵也不同

在此之前,总结出来一个规律:对于求矩阵中子矩阵的面积/大小/和等问题,其实都可以转换为一维数组前缀和的问题.
通过枚举遍历的方式

for (int i=0;i<matrix.length;i++){
            int[] preSum = new int[matrix[0].length];//计算每一列的前缀和
            for (int j=i;j<matrix.length;j++){
                for (int k=0;k<matrix[0].length;k++){
                    preSum[k] = preSum[k]+matrix[j][k];
                }
                ans+=subarraySum(preSum,target);
            }
        }

仔细看:
第一个for (int i=0;i<matrix.length;i++)循环其实就是子矩阵的上边界
第二个for (int j=i;j<matrix.length;j++)循环是子矩阵的下边界
因此就是从上到下,先遍历枚举整个矩阵,然后再枚举除第一行外的矩阵,之后是除第二行外的矩阵,一直到最后一行.
然后通过在矩阵变换的过程中求每一列的前缀和,通过前缀和就可以得到对应子矩阵的面积/大小/和等问题.

有了刚才这个总结之后,这个二维问题就可以转换为求一维数组中满足和为target的子序列个数问题.即和为K的子数组

• 枚举子矩阵的上下边界，并计算出该边界内每列的元素和，则原问题转换成了如下一维问题：给定一个整数数组和一个整数target，计算该数组中子数组和等于target 的子数组个数。

public static int numSubmatrixSumTarget(int[][] matrix, int target) {
        int ans = 0;

        for (int i=0;i<matrix.length;i++){//上边界
            int[] preSum = new int[matrix[0].length];//计算每一列的前缀和
            for (int j=i;j<matrix.length;j++){//下边界
                for (int k=0;k<matrix[0].length;k++){//更新每一列的和
                    preSum[k] = preSum[k]+matrix[j][k];
                }
                ans+=subarraySum(preSum,target);
            }
        }
        return ans;
    }

    //求一维数组中和为K的子数组
    public static int subarraySum(int[] nums, int k) {
        int ans = 0,preSum = 0;
        HashMap<Integer,Integer> map = new HashMap<>();
        map.put(0,1);
        for (int i=0;i<nums.length;i++){
            preSum+=nums[i];
            if (map.containsKey(preSum-k)){
                ans+=map.get(preSum-k);
            }
            map.put(preSum,map.getOrDefault(preSum,0)+1);
        }
        return ans;
    }