题目描述:
给你一个只包含 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 。
示例 2:
输入:mat = [[0,1,1,0],
[0,1,1,1],
[1,1,1,0]]
输出:24
解释:
有 8 个 1x1 的子矩形。
有 5 个 1x2 的子矩形。
有 2 个 1x3 的子矩形。
有 4 个 2x1 的子矩形。
有 2 个 2x2 的子矩形。
有 2 个 3x1 的子矩形。
有 1 个 3x2 的子矩形。
矩形数目总共 = 8 + 5 + 2 + 4 + 2 + 2 + 1 = 24 。
示例 3:
输入:mat = [[1,1,1,1,1,1]]
输出:21
示例 4:
输入:mat = [[1,0,1],[0,1,0],[1,0,1]]
输出:5
提示:
1 <= rows <= 150
1 <= columns <= 150
0 <= mat[i][j] <= 1
方法1:
主要思路:解题汇总链接
(1)使用二维数组 left_ones[ i ][ j ]辅助计算,表示原数组mat在位置 (i,j)开始,在第 i 行,从第 j列向前,有多少个连续的1;
(2)当再次遍历数组mat时,对于新的 (i,j)位置,新增的矩形数量可以是以(i,j)为矩形右下角的新的矩形的数量,则可以通过 left_ones[ k ][ j ] ,其中 k>=0&&k<=i,来计算可以新增的矩形的数量,进行统计;
class Solution {
public:
int numSubmat(vector<vector<int>>& mat) {
int rows=mat.size();
int cols=mat[0].size();
vector<vector<int>> left_ones(rows,vector<int>(cols,0));//统计原数组位置(i,j),在第 i 行,从第 j 列开始,向左连续的1的个数;
for(int i=0;i<rows;++i){
int cur=0;
for(int j=0;j<cols;++j){
if(mat[i][j]==0){
cur=0;
}
else{
++cur;
}
left_ones[i][j]=cur;
}
}
int res=0;//存储结果
for(int i=0;i<rows;++i){
for(int j=0;j<cols;++j){
int min_ones=INT_MAX;
for(int k=i;k>=0;--k){//以当前的left_ones[ i ][ j ]为低,可以形成的新增的矩形的数量
min_ones=min(min_ones,left_ones[k][j]);
if(min_ones==0){//说明再向上也不能形成连续的矩形了,故可以终止统计;
break;
}
res+=min_ones;
}
}
}
return res;
}
};