用java代码统计全为 1 的正方形子矩阵给你一个 m n 的矩阵矩阵中的元素不是 0 就是 1请你统计并返回其中完全由 1 组成的 正方形 子矩阵的个数。

可以使用动态规划来解决这个问题。

设 dp(i, j) 表示以 (i, j) 为右下角的最大正方形边长，则有状态转移方程：

dp(i, j) = min(dp(i-1, j), dp(i, j-1), dp(i-1, j-1)) + 1，当 matrix(i-1, j-1) == 1

其中，matrix(i-1, j-1) 表示矩阵中第 i 行第 j 列的元素。

这个方程的意思是，如果 (i, j) 为右下角的矩阵元素为 1，则它能够组成的最大正方形边长就是它上方、左方和左上方的三个元素能够组成的最大正方形边长的最小值再加上 1。

最后，统计所有 dp(i, j) 的和即可得到完全由 1 组成的正方形子矩阵的个数。

Java代码如下：

public int countSquares(int[][] matrix) { int m = matrix.length; int n = matrix[0].length; int[][] dp = new int[m+1][n+1]; int count = 0; for (int i = 1; i <= m; i++) { for (int j = 1; j <= n; j++) { if (matrix[i-1][j-1] == 1) { dp[i][j] = Math.min(dp[i-1][j], Math.min(dp[i][j-1], dp[i-1][j-1])) + 1; count += dp[i][j]; } } } return count; }