Armstrong Number – Program in Java and Python

• 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 — all single digit numbers are Armstrong.
• 153 (3 digits): 1³+5³+3³ = 153 ✓
• 370 (3 digits): 3³+7³+0³ = 27+343+0 = 370 ✓
• 371 (3 digits): 3³+7³+1³ = 371 ✓
• 407: 4³+0³+7³ = 64+0+343 = 407 ✓

num = int(input('Enter a number: '))
original = num
result = 0
n = len(str(num))  # number of digits
while num > 0:
    digit = num % 10
    result += digit ** n
    num //= 10
if result == original:
    print(f'{original} is an Armstrong number')
else:
    print(f'{original} is NOT an Armstrong number')

import java.util.Scanner;
public class Armstrong {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int num = sc.nextInt();
        int original = num, result = 0;
        int n = String.valueOf(num).length();
        while (num > 0) {
            int digit = num % 10;
            result += Math.pow(digit, n);
            num /= 10;
        }
        if (result == original)
            System.out.println(original + " is Armstrong");
        else
            System.out.println(original + " is not Armstrong");
    }
}