Leetcode 1314.矩阵区域和（Matrix Block Sum）

Leetcode 1314.矩阵区域和

1 题目描述（Leetcode题目链接）

  给你一个 $m\ast n$ 的矩阵 mat 和一个整数 K ，请你返回一个矩阵 answer ，其中每个 $answer[i][j]$ 是所有满足下述条件的元素 $mat[r][c]$ 的和：

• $i-K\le r\le i+K,j-K\le c\le j+K$
• $(r,c)$ 在矩阵内。

输入：mat = [[1,2,3],[4,5,6],[7,8,9]], K = 1
输出：[[12,21,16],[27,45,33],[24,39,28]]

输入：mat = [[1,2,3],[4,5,6],[7,8,9]], K = 2
输出：[[45,45,45],[45,45,45],[45,45,45]]

提示：

• m == mat.length
• n == mat[i].length
• 1 <= m, n, K <= 100
• 1 <= mat[i][j] <= 100

2 题解

  先记录矩阵前缀和，多加个一行一列方便处理边界，然后根据每个点计算它的左上边界和右下边界再计算answer的值，具体可参考第304题。

class Solution:
    def matrixBlockSum(self, mat: List[List[int]], K: int) -> List[List[int]]:
        m = len(mat)
        n = len(mat[0])
        matrix = [([0]*(n+1)) for _ in range(m+1)]
        res = [([0]*n) for _ in range(m)]
        for i in range(1, m+1):
            for j in range(1, n+1):
                matrix[i][j] = matrix[i-1][j] + matrix[i][j-1] + mat[i-1][j-1] - matrix[i-1][j-1]

        for i in range(m):
            for j in range(n):
                row1, col1 = max(i - K, 0), max(j - K, 0)
                row2, col2 = min(i + K, m - 1), min(j + K, n - 1)
                res[i][j] = matrix[row2+1][col2+1] - matrix[row2+1][col1] - matrix[row1][col2+1] + matrix[row1][col1]
        return res