每日OJ题_算法_前缀和⑧_力扣1314. 矩阵区域和

目录

[力扣1314. 矩阵区域和](#力扣1314. 矩阵区域和)

解析代码


力扣1314. 矩阵区域和

1314. 矩阵区域和

难度 中等

给你一个 m x 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
cpp 复制代码
class Solution {
public:
    vector<vector<int>> matrixBlockSum(vector<vector<int>>& mat, int k) {

    }
};

解析代码

题意有点类似C语言写过的三子棋的一个功能?要返回的二维数组就是原数组上下左右移动K的单位包起来的正方形区域所有元素的和(越界的不管),用二维前缀和思路解决。

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>> dp(m + 1, vector<int>(n + 1));
        for(int i = 1; i <= m; ++i)
            for(int j = 1; j <= n; ++j) // 此题的mat下标不是从1开始的,注意转化
                dp[i][j] = dp[i-1][j] + dp[i][j-1] - dp[i-1][j-1] + mat[i-1][j-1];
        
        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; // 左上角
                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;   
    }
};
相关推荐
我是why的狗几秒前
赵义弘-----补题报告
算法·排序算法
源代码•宸18 分钟前
深入浅出设计模式——创建型模式之建造者模式 Builder
c++·经验分享·设计模式·建造者模式
roman_日积跬步-终至千里1 小时前
【机器学习】两大线性分类算法:逻辑回归与线性判别分析:找到分界线的艺术
算法·机器学习·分类
YuTaoShao1 小时前
【LeetCode 热题 100】739. 每日温度——(解法一)单调栈+从右到左
java·算法·leetcode·职场和发展
itgather1 小时前
智能图书馆管理系统开发实战系列(五):前后端集成 - koffi调用与接口设计
c++·koffi
Spider_Man1 小时前
栈中藏玄机:从温度到雨水,单调栈的逆袭之路
javascript·算法·leetcode
爱吃KFC的大肥羊2 小时前
C/C++常用字符串函数
c语言·数据结构·c++·算法
im_AMBER2 小时前
Leetcode 11 java
java·算法·leetcode
丶小鱼丶2 小时前
二叉树算法之【Z字型层序遍历】
java·算法
岁忧4 小时前
(nice!!!)(LeetCode 每日一题) 2561. 重排水果 (哈希表 + 贪心)
java·c++·算法·leetcode·go·散列表