矩阵区域和 --- 前缀和

目录

一:题目

二:算法原理

三:代码


一:题目

题目链接:1314. 矩阵区域和 - 力扣(LeetCode)

二:算法原理

三:代码

cpp 复制代码
class Solution {
public:
    vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) 
    {
        int m = mat.size(),n = mat[0].size();
        vector<vector<int>> ret(m,vector<int>(n));
        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];
            }
        } 

        //使用前缀和
        for(int i = 0; i < m ;i++)
        {
            for(int j = 0; j < n ; j++)
            {
                int x1 = max(0,i-k)+1;
                int y1 = max(0,j-k)+1;
                int x2 = min(m-1,i+k)+1;
                int y2 = min(n-1,j+k)+1;

                ret[i][j] = dp[x2][y2] - dp[x1-1][y2]-dp[x2][y1-1]+dp[x1-1][y1-1];
            }
        }
        
        return ret;
    }
};
相关推荐
姜暮儿1 小时前
C++ 性能优化
开发语言·c++
菜鸟555552 小时前
图论:最小生成树
算法·图论
铭哥的编程日记3 小时前
《从C风格到C++风格:内存管理的进化之路》
开发语言·c++
2401_872945093 小时前
【补题】Codeforces Round 735 (Div. 2) C. Mikasa
算法
叫我:松哥3 小时前
基于网络爬虫的在线医疗咨询数据爬取与医疗服务分析系统,技术采用django+朴素贝叶斯算法+boostrap+echart可视化
人工智能·爬虫·python·算法·django·数据可视化·朴素贝叶斯
Star在努力3 小时前
14-C语言:第14天笔记
c语言·笔记·算法
程序员编程指南4 小时前
Qt 与 SQLite 嵌入式数据库开发
c语言·数据库·c++·qt
赴3355 小时前
Numpy 库 矩阵数学运算,点积,文件读取和保存等
人工智能·算法·numpy·random·dot
自由随风飘5 小时前
机器学习-SVM支持向量机
算法·机器学习·支持向量机
屁股割了还要学6 小时前
【C语言进阶】柔性数组
c语言·开发语言·数据结构·c++·学习·算法·柔性数组