统计全为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;}
}