首页 > 其他 > 详细

1074. 元素和为目标值的子矩阵数量

时间:2021-05-30 00:35:08      阅读:25      评论:0      收藏:0      [点我收藏+]

给出矩阵 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‘),那么这两个子矩阵也不同。

 

/**
 * @param {number[][]} matrix
 * @param {number} target
 * @return {number}
 */
var numSubmatrixSumTarget = function(matrix, target) {
    let m = matrix.length;
    let n = matrix[0].length;
    let res = 0;
    for(let i = 0;i<m;i++){
        let sum = new Array(n).fill(0);
        for(let j=i;j<m;j++){
            for(let k=0;k<n;k++){
                sum[k] +=matrix[j][k];
            }
            res+=findSumTarget(sum,target);
        }
    }
    function  findSumTarget (sum,k) {
        let map = new Map();
        map.set(0,1);
        let pre = 0;
        let count = 0;
        for(num of sum){
            pre+=num;
            if(map.has(pre-k)){
                count+= map.get(pre-k)
            }
            map.set(pre, (map.get(pre)||0) + 1)
        }
        return count;
    }
    return res;
};

let matrix = [[0,1,0],[1,1,1],[0,1,0]], target = 0
console.log(matrix, target, numSubmatrixSumTarget(matrix,target))
matrix = [[1,-1],[-1,1]], target = 0
console.log(matrix, target, numSubmatrixSumTarget(matrix,target))
matrix = [[904]], target = 0
console.log(matrix, target, numSubmatrixSumTarget(matrix,target))

示例 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 子矩阵。

示例 3:

输入:matrix = [[904]], target = 0
输出:0

 

提示:

  • 1 <= matrix.length <= 100
  • 1 <= matrix[0].length <= 100
  • -1000 <= matrix[i] <= 1000
  • -10^8 <= target <= 10^8

1074. 元素和为目标值的子矩阵数量

原文:https://www.cnblogs.com/yanjianjiang/p/14826687.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!