給你一個(gè) m * n的矩陣mat和一個(gè)整數(shù) K，請(qǐng)你返回一個(gè)矩陣answer ，其中每個(gè)answer[i][j] 是所有滿足下述條件的元素 mat[r][c] 的和：
i - K <= r <= i + K, j - K <= c <= j + K
(r, c)在矩陣內(nèi)。

示例 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

來(lái)源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/matrix-block-sum

圖片來(lái)源網(wǎng)上

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; }
}

總結(jié)：預(yù)處理