矩阵区域和

矩阵区域和

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

还是前缀和的题目,先计算矩阵的前缀和,然后修订求解区域的上下界,之后直接得到该区域的和即可

//先计算前缀和 然后将上界下界进行修订 进行区域内和的计算即可
    public static int[][] matrixBlockSum(int[][] mat, int k) {
        int n = mat.length;
        int m = mat[0].length;
        int[][] sum = new int[n + 1][m + 1];//求出矩阵的前缀和

        //计算出从最左上角一直到mat[i][j]所有值的和
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                sum[i][j] = mat[i - 1][j - 1] + sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1];
            }
        }

        int[][] ans = new int[n][m];
        int leftRow, leftCol, rightRow, rightCol;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                //不能超过矩阵的界(将上界下界进行修订)
                leftRow = i - k < 0 ? 0 : i - k;
                leftCol = j - k < 0 ? 0 : j - k;
                rightRow = i + k > n - 1 ? n - 1 : i + k;
                rightCol = j + k > m - 1 ? m - 1 : j + k;
                ans[i][j] = sum[rightRow+1][rightCol+1]+sum[leftRow][leftCol]
                        -sum[leftRow][rightCol+1]-sum[rightRow+1][leftCol];
            }
        }
        return ans;
    }