Java实现 LeetCode1314.矩阵区域和（动态规划）

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

i - K <= r <= i + K, j - K <= c <= j + K
(r, c) 在矩阵内。

示例 1：

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

输入：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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/matrix-block-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解：
这道题主要分两大部分，涉及到前缀和。

1. 二维数组sum，sum[i][j] 表示,mat 的 [0][0] 到[i][j]位置的和。
   sum[i][j] = sum[i-1][j]+sum[i][j-1] + mat[i-1][j-1] - sum[i-1][j-1]
   为了方便计算我们sum的下标从1开始。画图对这个前缀和计算会更清晰。
2. answer数组，answer[i-1][j-1] = sum[i+K+1][j+K+1] - sum[i-k][j+k+1]-sum[i+K+1][j-K] + sum[i-K][j-K]
   我们还需要处理越界i+K, i-K，j+K,j-K 的越界情况。

class Solution {
    public int[][] matrixBlockSum(int[][] mat, int K) {
        int m = mat.length;
        int n = mat[0].length;
        int[][] sum = new int[m+1][n+1];
        int[][] answer = new int[m][n];
        //1. 求前缀和
        for(int i = 1; i <= m; i++) {
            for(int j = 1; j <= n; j++) {
                sum[i][j] = sum[i-1][j] + sum[i][j-1] + mat[i-1][j-1] - sum[i-1][j-1];
            }
        }

        //求answer
        for(int i = 1; i <= m; i++) {
            for(int j = 1; j <= n; j++) {
                //需要处理一下边界情况
                int x0 = Math.max(0, i -  K - 1);
                int y0 = Math.max(0, j - K - 1);
                int x1 = Math.min(m,i + K);
                int y1 = Math.min(n, j + K);

                answer[i - 1][j - 1] = sum[x1][y1] - sum[x1][y0] - sum[x0][y1] + sum[x0][y0];
            }
        }
        return answer;
    }
}