给你一个只包含 0 和 1 的 rows * columns 矩阵 mat ,请你返回有多少个 子矩形 的元素全部都是 1 。
示例 1:
输入:mat =
[[1,0,1],
[1,1,0],
[1,1,0]]
输出:13
解释:
有 6 个 1x1 的矩形。
有 2 个 1x2 的矩形。
有 3 个 2x1 的矩形。
有 1 个 2x2 的矩形。
有 1 个 3x1 的矩形。
矩形数目总共 = 6 + 2 + 3 + 1 + 1 = 13 。
解题思路
数组含义:dp[i][j]位于(i,j)的元素向左延长的长度
状态转移:min= Math.min(dp[k][j],min) 向上遍历,加入满足最小长度的矩形
代码
class Solution {public int numSubmat(int[][] mat) {int n=mat.length,m=mat[0].length,res=0;int[][]dp=new int[n][m+1];for(int i=0;i<n;i++)for(int j=1;j<=m;j++)dp[i][j]=mat[i][j-1]==1?dp[i][j-1]+1:0;for(int i=0;i<n;i++)for(int j=1;j<=m;j++){int min=Integer.MAX_VALUE;for (int k=i;k>=0;k--)if(dp[k][j]==0) break;else{min= Math.min(dp[k][j],min);res+=min;}}return res;}
}