leet-code/src/main/java/leetcode/editor/cn/NumberOfSubmatricesThatSumToTarget.java

93 lines
2.6 KiB
Java
Raw Normal View History

//给出矩阵 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'),那么这两个子矩阵
//也不同。
//
//
//
// 示例 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
//
// Related Topics 数组 动态规划 Sliding Window
// 👍 117 👎 0
package leetcode.editor.cn;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
//1074:元素和为目标值的子矩阵数量
class NumberOfSubmatricesThatSumToTarget {
public static void main(String[] args) {
//测试代码
Solution solution = new NumberOfSubmatricesThatSumToTarget().new Solution();
System.out.println(solution.numSubmatrixSumTarget(new int[][]{{0,1,0}, {1,1,1},{0,1,0}}, 0));
}
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int numSubmatrixSumTarget(int[][] matrix, int target) {
int xLength = matrix.length;
int yLength = matrix[0].length;
int[][] sum = new int[xLength + 1][yLength + 1];
for (int i = 1; i <= xLength; ++i) {
for (int j = 1; j <= yLength; ++j) {
sum[i][j] = matrix[i - 1][j - 1] + sum[i - 1][j] + sum[i][j - 1] - sum[i - 1][j - 1];
}
}
int count = 0;
for (int i = 1; i <= xLength; ++i) {
for (int j = 1; j <= yLength; ++j) {
for (int k = 1; k <= i; ++k) {
for (int l = 1; l <= j; ++l) {
if (sum[i][j] - sum[k - 1][j] - sum[i][l - 1] + sum[k - 1][l - 1] == target){
++count;
}
}
}
}
}
return count;
}
}
//leetcode submit region end(Prohibit modification and deletion)
}