統計全為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;}
}