Matrix Block Sum（C++矩阵区域和）

女爷i 2023-01-02 03:25 87阅读 0赞

解题思路：

（1）dp数组，防止重复计算累加和

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,vector<int>(n,0)),v(m,vector<int>(n,0));
            
            for(int i=0;i<m;i++) {
                for(int j=0;j<n;j++) {
                    if(i==0 && j==0) dp[i][j]=mat[i][j];
                    else if(i==0 && j>0) dp[i][j]=dp[i][j-1]+mat[i][j];
                    else if(i>0 && j==0) dp[i][j]=dp[i-1][j]+mat[i][j];
                    else dp[i][j]=dp[i][j-1]+dp[i-1][j]-dp[i-1][j-1]+mat[i][j];
                }
            }
            
            for(int i=0;i<m;i++) {
                for(int j=0;j<n;j++) {
                    int i1=i-K-1,j1=j-K-1;
                    int i2=i+K>m-1?m-1:i+K;
                    int j2=j2=j+K>n-1?n-1:j+K;
                    if(i1<0 && j1<0) v[i][j]=dp[i2][j2];
                    else if(i1<0 && j1>=0) v[i][j]=dp[i2][j2]-dp[i2][j1];
                    else if(i1>=0 && j1<0) v[i][j]=dp[i2][j2]-dp[i1][j2];
                    else v[i][j]=dp[i2][j2]-dp[i1][j2]-dp[i2][j1]+dp[i1][j1];
                }
            }
            
            return v;
            
        }
    };

发表评论取消回复

表情：

评论列表（有 0 条评论，87人围观）

还没有评论，来说两句吧...

相关阅读

相关 Python：numpy矩阵库(Matrix)

本章内容了解 NumPy 矩阵库。NumPy数组属性详见[https://mp.csdn.net/console/editor/html/105042451][https_mp

野性酷女/ 2023年07月17日 15:58/ 0 赞/ 145 阅读

相关 Matrix Block Sum（C++矩阵区域和）

解题思路：（1）dp数组，防止重复计算累加和 class Solution { public: vector<vector<int>> m

女爷i/ 2023年01月02日 03:25/ 0 赞/ 88 阅读

相关 Spiral Matrix（C++螺旋矩阵）

(1) 设置访问数组+转向 class Solution { public: vector<vector<int>> d={

水深无声/ 2022年10月11日 00:51/ 0 赞/ 207 阅读

相关 Python Numpy的数组array和矩阵matrix

出处：[http://blog.chinaunix.net/uid-21633169-id-4408596.html][http_blog.chinaunix.net_uid-

た入场券/ 2022年06月16日 08:44/ 0 赞/ 259 阅读

相关 [Leetcode][python]Spiral Matrix/Spiral Matrix II/螺旋矩阵/螺旋矩阵 II

Spiral Matrix 题目大意将一个矩阵中的内容螺旋输出。注意点：矩阵不一定是正方形例子：输入: matrix = \[ \[1, 2,

旧城等待，/ 2022年06月07日 03:55/ 0 赞/ 213 阅读

相关黑塞矩阵（Hessian Matrix）

转载自http://blog.csdn.net/caimouse/article/details/60465975，仅用作个人学习。如需删除，请见谅并联系本人。为了不分原创浏览

╰+攻爆jí腚メ/ 2022年06月04日 09:42/ 0 赞/ 327 阅读

相关 confusion matrix 混淆矩阵绘制

原理横轴一般是predict label，纵轴是ground truth label，对角线是预测正确的概率或个数绘制 sklearn 中confusion\

淡淡的烟草味﹌/ 2022年05月12日 02:50/ 0 赞/ 315 阅读

相关 Confusion Matrix（混淆矩阵）

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ub

偏执的太偏执、/ 2022年03月01日 18:26/ 0 赞/ 301 阅读

相关 CSS3 matrix矩阵

参数 matrix()有六个参数：matrix(a,b,c,d,e,f)；这六个参数组成的矩阵与原坐标矩阵相乘计算坐标；计算 ![aHR0cHM6L

港控/mmm°/ 2021年10月18日 06:26/ 0 赞/ 408 阅读

相关拉普拉斯矩阵（Laplacian Matrix）

文章目录 1. 无向加权图 G G G 2. 邻接矩阵 W W W 3. 度矩阵 D D D 4. 拉普拉斯矩阵 L L L 参考

系统管理员/ 2021年09月25日 01:38/ 0 赞/ 2045 阅读