LeetCode 1314. 矩阵区域和

给你一个 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

图片来源网上
计算以[ i，j ]为中心的区域（红色区域）.

class Solution {
    public int[][] matrixBlockSum(int[][] mat, int K) {
        int m = mat.length,n = mat[0].length;
        int[][] res = new int[m][n];

        for(int i=0;i<m;i++){
            for(int j=1;j<n;j++)
                mat[i][j]+=mat[i][j-1];
        }

        for(int j=0;j<n;j++){
            for(int i=1;i<m;i++){
                mat[i][j]+=mat[i-1][j];
            }
        }

        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                res[i][j] += mat[min(i+K,m-1)][min(j+K,n-1)];
                if(i-K-1>=0) res[i][j] -= mat[i-K-1][min(j+K,n-1)];
                if(j-K-1>=0) res[i][j] -= mat[min(i+K,m-1)][j-K-1];
                if(i-K-1>=0 && j-K-1>=0) res[i][j]+=mat[i-K-1][j-K-1];
            }
        }

        return res;
    }
    private int min(int a,int b){ return a>b?b:a; }
}

总结：预处理