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

示例 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 子矩阵。

示例 3：

输入：matrix = [[904]], target = 0
输出：0

提示：

1 <= matrix.length <= 100
1 <= matrix[0].length <= 100
-1000 <= matrix[i] <= 1000
-10^8 <= target <= 10^8

code:

def numSubmatrixSumTarget(matrix, target):
    m, n = len(matrix), len(matrix[0])
    # 求列的前缀和
    sum_ = [[0]*n]
    for i in range(1, m+1):
        sum_.append([])
        for j in range(0, n):
            sum_[i].append(sum_[i-1][j] + matrix[i-1][j])
    print(sum_)

    count = 0
    for i in range(1, m+1):  # 枚举上边界
        for j in range(i, m+1):  # 枚举下边界
            map_ = dict()
            map_[0] = 1
            s = 0
            for k in range(0, n):  # 枚举每个列元素
                s += sum_[j][k] - sum_[i-1][k]
                count += map_.get(s - target, 0)
                map_[s] = map_.get(s, 0) + 1

    return count

print(numSubmatrixSumTarget([[0,1,1,1,0,1],[0,0,0,0,0,1],[0,0,1,0,0,1],[1,1,0,1,1,0],[1,0,0,1,0,0]], 0))