统计全为1的正方形子矩阵
这题是正方形,比较简单
暴力就是二维前缀和。
或者用dp
,dp[i][j
]表示以i
,j
为右下角的最大正方形的边长,它由(i-1,j),(i,j-1),(i-1,j-1)
三者共同决定,通过找规律可知,由三个的最小值决定
class Solution {
public int countSquares(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
int[][] f = new int[m + 1][n + 1];
int ans = 0;
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++) {
if (matrix[i][j] > 0) {
f[i + 1][j + 1] = Math.min(Math.min(f[i][j], f[i][j + 1]), f[i + 1][j]) + 1;
ans += f[i + 1][j + 1];
}
}
}
return ans;
}
}