leetcode-1074

问题描述

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

思路

• 前缀和的思路，就是计算前 $x，y$ 的和 $s[x][y]$。 然后利用 s 计算子矩阵 $M(x_1,y_1,x_2,y_2)$ 的和，但是上面结果的复杂度为 $O(n^4)$。
• 使用其他的方法
  • 分开求取，使用$S[x][y]$ 表示第 x 行的前 y 列数据之和
  • $rowdict[j1][j2][k]$ 表示 列的边界为 $j1$ 到$j2$ 的行的前缀和（前 k 行的和）
  • 用一个字典来保存 $rowdict[j1][j2][k]$值出现的次数
  • （j1，j2固定的情况下）如果当前在第k 行，且前k行的数据 cur =100，目标值 target=80，那么我们就需要寻找之前的出现过 20 的次数。出现20的行号 k1 ，到k 的值和就为 target。

class Solution(object):
    def numSubmatrixSumTarget(self, matrix, target):
        """
        :type matrix: List[List[int]]
        :type target: int
        :rtype: int
        """
        ## 前缀和的概念
        m = len(matrix)
        n = len(matrix[0])
        # print(m,n)
        # 先计算前缀和
        maps = [[0]*n for _ in range(m)]
        maps[0][0] = matrix[0][0]
        for i in range(m):
            maps[i][0] = matrix[i][0]
            for j in range(1,n):
                maps[i][j] = maps[i][j-1]+matrix[i][j]
        # print(maps)
        ##利用 pre 求取结果 j1，j2 列的前行的和
        res= 0
        for i in range(n):
            for j in range(i,n):
            ## 使用字典保存 pre[j1][j2][k]  出现的数
                rowdict = collections.defaultdict(int)
                cur = 0
                rowdict[0] = 1
                for k in range(m):
                #cur表示 前k 行 从第 i列到 j列的数字和
                # map[k][j]:表示第k行从 i 列到j列的数字和
                    cur+=maps[k][j]-(maps[k][i-1] if i>0 else 0)
                    res+=rowdict[cur - target]
                    rowdict[cur]+=1
        
        return res