LeetCode 1314 矩阵区域和

最新推荐文章于 2022-12-06 13:11:23 发布

iNiegang

最新推荐文章于 2022-12-06 13:11:23 发布

阅读量572

点赞数

分类专栏： java 算法文章标签： leetcode 算法

本文链接：https://blog.youkuaiyun.com/iNiegang/article/details/104101743

版权

java 同时被 2 个专栏收录

31 篇文章

订阅专栏

算法

25 篇文章

订阅专栏

该博客介绍了LeetCode上的第1314题，即矩阵区域和的问题。内容包括问题描述、解题思路和代码实现。博主提供了详细的解决方案，解释了如何计算满足特定条件的矩阵元素之和，并给出了相应的Python代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题描述

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