前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1074. 元素和为目标值的子矩阵数量(2d前缀和+哈希)

LeetCode 1074. 元素和为目标值的子矩阵数量(2d前缀和+哈希)

作者头像
Michael阿明
发布2021-02-19 14:39:27
3980
发布2021-02-19 14:39:27
举报
文章被收录于专栏:Michael阿明学习之路

文章目录

1. 题目

给出矩阵 matrix 和目标值 target,返回元素总和等于目标值的非空子矩阵的数量。

子矩阵 x1, y1, x2, y2 是满足 x1 <= x <= x2 且 y1 <= y <= y2 的所有单元 matrix[x][y] 的集合。

如果 (x1, y1, x2, y2)(x1', y1', x2', y2') 两个子矩阵中部分坐标不同(如:x1 != x1'),那么这两个子矩阵也不同。

代码语言:javascript
复制
示例 1:
输入:matrix = 
[[0,1,0],
 [1,1,1],
 [0,1,0]], 
 target = 0
输出:4
解释:四个只含 0 的 1x1 子矩阵。

示例 2:
输入:matrix = 
[[1,-1],
 [-1,1]], 
 target = 0
输出:5
解释:两个 1x2 子矩阵,加上两个 2x1 子矩阵,再加上一个 2x2 子矩阵。
 
提示:
1 <= matrix.length <= 300
1 <= matrix[0].length <= 300
-1000 <= matrix[i] <= 1000
-10^8 <= target <= 10^8

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/number-of-submatrices-that-sum-to-target 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 先递推求出,左上角(0,0)到(i, j)的区域的和
  • 然后枚举两个 行号,一个列号,求取两个行夹住的区域的前缀和,利用哈希记录前缀和出现的次数
  • 时间复杂度 O ( m 2 n ) O(m^2n) O(m2n)
代码语言:javascript
复制
class Solution {
public:
    int numSubmatrixSumTarget(vector<vector<int>>& matrix, int target) {
        int m = matrix.size(), n = matrix[0].size();
        vector<vector<int>> presum(m+1, vector<int>(n+1, 0));
        for(int i = 1; i <= m; i++) 
        {
            for(int j = 1; j <= n; j++)
            {
                presum[i][j] = matrix[i-1][j-1]+presum[i-1][j]+presum[i][j-1]-presum[i-1][j-1];
            }
        }// 左上角 [0,0] 到 右下角[i,j] 的 前缀和
        int ans = 0;
        for(int i1 = 1; i1 <= m; i1++)
        {
            for(int i2 = i1; i2 <= m; i2++)
            {   //枚举两个顶点的 行的 idx 组合
                unordered_map<int,int> map; // 和,计数
                for(int j = 1; j <= n; j++)
                {
                    int sum = presum[i2][j]-presum[i1-1][j];//两行夹住的区域
                    if(sum == target)
                        ans++;
                    if(map.find(sum-target) != map.end())
                    {
                        ans += map[sum-target];
                    }
                    map[sum]++;
                }
            }
        }
        return ans;
    }
};

1604 ms 158 MB C++


本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2021/01/02 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 文章目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档