Upload files to "OOP/Java/Lab/Week3"

OOP/Java/Lab/Week3/Armstrong.java

@@ -1,4 +1,5 @@
 import java.util.Scanner;
+import java.lang.Math;
 
 class Armstrong{
     public static void main(String args[]){
@@ -6,15 +7,23 @@ class Armstrong{
 
         int a,b,digit;
         int digisum=0;
+        int c=0;
 
         System.out.println("Enter the number you went to check for Armstrong-ness:");
         a = sc.nextInt();
+    
+        b = a;
+
+        while(b!=0){
+            c++;
+            b/=10;
+        }
 
         b = a;
 
         while(b!=0){
             digit = b%10;
-            digisum += (digit*digit*digit);
+            digisum += Math.pow(digit, c);
             b /= 10;
         }