public int countSquares(int[][] matrix) {
    int count = 0;//正方形的个数
    int[][] dp = new int[matrix.length + 1][matrix[0].length + 1];
    for (int i = 0; i < matrix.length; i++) {
        for (int j = 0; j < matrix[0].length; j++) {
            //如果当前坐标是0，就不可能构成正方形，直接跳过
            if (matrix[i][j] == 0)
                continue;
            //递推公式
            dp[i + 1][j + 1] = Math.min(Math.min(dp[i + 1][j], dp[i][j + 1]), dp[i][j]) + 1;
            //累加所有的dp值
            count += dp[i + 1][j + 1];
        }
    }
    return count;
}

总结

搞懂了《530，动态规划解最大正方形》，这题就非常简单了，这里需要理解的是以坐标(i,j)为右下角的最大正方形边长就是以(i,j)为右下角正方形的个数。