矩阵区域和 --- 前缀和

目录

一:题目

二:算法原理

三:代码


一:题目

题目链接: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;
    }
};
相关推荐
桐果云5 分钟前
解锁桐果云零代码数据平台能力矩阵——赋能零售行业数字化转型新动能
大数据·人工智能·矩阵·数据挖掘·数据分析·零售
深耕AI10 分钟前
【MFC中OnInitDialog虚函数详解:哪个是虚函数?两个OnInitDialog的关系】
c++·mfc
CHANG_THE_WORLD33 分钟前
并发编程指南 同步操作与强制排序
开发语言·c++·算法
pl002038 分钟前
C++虚函数&虚析构函数&纯虚函数的使用说明和理解
c++·虚函数·纯虚函数·虚析构函数
gaoshou452 小时前
代码随想录训练营第三十一天|LeetCode56.合并区间、LeetCode738.单调递增的数字
数据结构·算法
自信的小螺丝钉2 小时前
Leetcode 240. 搜索二维矩阵 II 矩阵 / 二分
算法·leetcode·矩阵
闪电麦坤952 小时前
数据结构:深度优先搜索 (Depth-First Search, DFS)
数据结构·深度优先
KING BOB!!!3 小时前
Leetcode高频 SQL 50 题(基础版)题目记录
sql·mysql·算法·leetcode
小wanga3 小时前
C++知识
java·开发语言·c++
深思慎考3 小时前
LinuxC++项目开发日志——高并发内存池(1-定长内存池)
linux·c++