给你一个 m x n 的二进制矩阵 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 。
提示:
1 <= m, n <= 150
mat[i][j] 仅包含 0 或 1
法一:枚举
class Solution {
public:
int numSubmat(vector<vector<int>>& mat) {
int m = mat.size(), n = mat[0].size();
vector<vector<int>> dp(m, vector<int>(n));
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
if(j == 0){
dp[i][j] = mat[i][j];
}
else if(mat[i][j] != 0){
dp[i][j] = dp[i][j-1] + 1;
}
else{
dp[i][j] = 0;
}
}
}
int ans = 0;
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
int col = dp[i][j];
for(int k = i; k >=0 && col; k--){
col = min(dp[k][j], col);
ans += col;
}
}
}
return ans;
}
};
时间复杂度:O(nm^2)
空间复杂度:O(mn)
这道题要求的是子矩形,那么一个矩形如何确定,实际上只要知道两个顶点就可以确定一个矩形。所以我们可以想到遍历矩阵每个元素作为矩形的右下角,然后来找左上顶点。既然右下角顶点已经确定,假设当矩形的高也知道,那么顶边的长度就是符合右下角为顶点,高确定的矩形个数。
所以我们要做的就是遍历矩阵每个元素为矩形右下角,然后在不断假设高,然后找到在不同高的情况下矩形的顶边最长是多少。
所以为了找到顶边,所以开始使用的dp就是为了计算在某一格元素向左的最大边长是多少。
我们将每个元素为右下角能构成的矩形数量都计入ans中,最后返回的ans就是所有符合要求全为1的子矩阵数量。
这道题还有单调栈优化,后面有时间补充