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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

• 1、首先要看清楚题目要求：就是求矩阵前缀和的
• 2、两层遍历result矩阵，对每一个元素去mat矩阵中获取数据并求和即可

代码实现

public int[][] matrixBlockSum(int[][] mat, int K) {
        int lineNum = mat.length;
        int colNum = mat[0].length;
        int[][] result = new int[lineNum][colNum];

        for (int resultLineNo = 0; resultLineNo < lineNum; resultLineNo++) {
            for (int resultColNo = 0; resultColNo < colNum; resultColNo++) {
                for (int matLine = resultLineNo - K; matLine <= resultLineNo + K && matLine < lineNum; matLine++) {
                    for (int matCol = resultColNo - K; matCol <= resultColNo + K && matCol < colNum; matCol++) {
                        if (matLine < 0 || matCol < 0) {
                            continue;
                        }
                        result[resultLineNo][resultColNo] += mat[matLine][matCol];
                    }
                }
            }
        }
        return result;
    }