Optimized Armstrong number code

ArmstrongNumber.java

@@ -0,0 +1,30 @@
+import java.util.Scanner;
+
+public class ArmstrongNumber {
+    public static void main(String[] args) {
+        Scanner sc = new Scanner(System.in);
+        System.out.print("Enter a number: ");
+        int num = sc.nextInt();
+        sc.close();
+
+        int originalNum = num;
+        int sum = 0;
+
+        // Find the number of digits
+        int digits = String.valueOf(num).length();
+
+        // Calculate sum of powers of digits
+        while (num > 0) {
+            int digit = num % 10;
+            sum += Math.pow(digit, digits);
+            num /= 10;
+        }
+
+        // Check Armstrong condition
+        if (sum == originalNum) {
+            System.out.println(originalNum + " is an Armstrong number.");
+        } else {
+            System.out.println(originalNum + " is NOT an Armstrong number.");
+        }
+    }
+}