Chương trình kiểm tra số Armstrong (Narcissistic) trong Python

Cho số n, kiểm tra xem có bằng tổng các chữ số mũ số chữ số hay không (số Armstrong).

Ví dụ

Đầu vào: 9474.
Đầu ra: True (9^4+4^4+7^4+4^4 = 9474).

Thuật toán

1. Chuyển n thành chuỗi s.
2. Tính tổng sum(int(x)**len(s) for x in s).
3. Trả về True nếu tổng bằng n.

Mã nguồn Python

class Solution:
    def solve(self, n):
        s = str(n)
        return n == sum(int(x)**len(s) for x in s)

ob = Solution()
print(ob.solve(9474))

Đầu vào / Đầu ra

Input: 9474
Output: True