Leetcode每日一题】 二维前缀和 - 矩阵区域和(难度⭐⭐)(31)

1. 题目解析

题目链接:1314. 矩阵区域和

题目乍一看很晦涩难懂,又是大于等于又是k的

仔细分析,题目所说的意思就是以[i,j]为中心,求上下左右向外拓展k个单位的矩阵和,放在名为answer的矩阵中,最后返回answer矩阵即可

2.算法原理

在填写结果矩阵的过程中,一个核心步骤是确定原矩阵中对应区域的"左上角"和"右下角"的坐标。为了更好地理解,建议大家通过画图来辅助。

  • 首先,我们考虑左上角的坐标。理论上,它的x坐标是i减去k,y坐标是j减去k。但是,由于坐标可能超出原矩阵的范围,我们必须确保坐标不会小于0。因此,我们需要对0取最大值。经过修正,左上角的坐标为:x1 = max(0, i - k),y1 = max(0, j - k)。
  • 接下来,我们确定右下角的坐标。理论上,它的x坐标是i加上k,y坐标是j加上k。但同样地,我们也要考虑坐标不能超出原矩阵的边界,即不能大于m-1和n-1。因此,我们需要对m-1和n-1取最小值。修正后的右下角坐标为:x2 = min(m - 1, i + k),y2 = min(n - 1, j + k)。
  • 一旦我们得到了这两个关键坐标,就可以将它们代入到二维前缀和矩阵的计算公式中。不过,需要注意的是,在代入坐标时,我们可能需要考虑到下标的映射关系,确保计算的准确性。

通过这样的步骤,我们可以有效地确定原矩阵中对应区域的边界,进而进行后续的计算和填写结果矩阵的操作。

3.代码编写

复制代码
class Solution 
{
public:
    vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) 
    {
        int m = mat.size(), n = mat[0].size();
        vector<vector<int>> dp(m + 1, vector<int>(n + 1));
        for(int i = 1; i <= m; i++)
            for(int j =1; j <= n; j++)
                dp[i][j] = dp[i - 1][j] + dp[i][j - 1] - dp[i - 1][j - 1] + mat[i - 1][j - 1];//减一为了匹配原来的mat
        vector<vector<int>> ret(m, vector<int>(n));
        for(int i = 0; i < m; i++)
            for(int j = 0; j < n; j++)
            {
                int x1 = max(0, i - k) + 1, y1 = max(0, j - k) + 1;
                int x2 = min(m - 1, i + k) + 1, y2 = min(n - 1, j + k) + 1;//加一为了匹配dp表
                ret[i][j] = dp[x2][y2] - dp[x1 - 1][y2] - dp[x2][y1 - 1] + dp[x1 - 1][y1 - 1];
            }
        return ret;
    }
};

写在最后

嗯,就是这样啦,文章到这里就结束啦,真心感谢你花时间来读。

觉得有点收获的话,不妨给我点个吧!

如果发现文章有啥漏洞或错误的地方,欢迎私信我或者在评论里提醒一声~

相关推荐
Lester_11011 小时前
嵌入式学习笔记 - 用泰勒公式解决 tanh函数
笔记·学习·算法
软件测试媛1 小时前
14:00开始面试,14:06就出来了,问的问题有点变态。。。
面试·职场和发展
无限进步_2 小时前
C语言字符串连接实现详解:掌握自定义strcat函数
c语言·开发语言·c++·后端·算法·visual studio
凤年徐2 小时前
HashMap 的哈希算法与冲突解决:深入 Rust 的高性能键值存储
算法·rust·哈希算法
J_Xiong01172 小时前
【VLNs篇】11:Dynam3D: 动态分层3D令牌赋能视觉语言导航中的VLM
人工智能·算法·3d
弈风千秋万古愁2 小时前
【PID】连续PID和数字PID chapter1(补充) 学习笔记
笔记·学习·算法·matlab
天选之女wow2 小时前
【代码随想录算法训练营——Day52】图论——101.孤岛的总面积、102.沉没孤岛、103.水流问题、104.建造最大岛屿
算法·深度优先·图论
碧海银沙音频科技研究院2 小时前
i2s封装成自己定义8路音频数据发送方法
arm开发·人工智能·深度学习·算法·音视频
做科研的周师兄2 小时前
【机器学习入门】9.2:感知机的工作原理 —— 从模型结构到实战分类
人工智能·算法·机器学习·分类·数据挖掘
不去幼儿园3 小时前
【启发式算法】狼群算法(Wolf Pack Algorithm, WPA)算法详细介绍(Python)
python·算法·启发式算法·任务分配·集群智能