LeetCode-Python-1134. 阿姆斯特朗数

假设存在一个 k 位数 N，其每一位上的数字的 k 次幂的总和也是 N，那么这个数是阿姆斯特朗数。

给你一个正整数 N，让你来判定他是否是阿姆斯特朗数，是则返回 true，不是则返回 false。

示例 1：

输入：153
输出：true
示例： 
153 是一个 3 位数，且 153 = 1^3 + 5^3 + 3^3。

示例 2：

输入：123
输出：false
解释： 
123 是一个 3 位数，且 123 != 1^3 + 2^3 + 3^3 = 36。

提示：

1. 1 <= N <= 10^8

思路：

依然水题，难度低于NOIP 普及组……

按要求把每一位的数字提出来然后计算K次幂之和即可。

class Solution(object):
    def isArmstrong(self, N):
        """
        :type N: int
        :rtype: bool
        """
        k = len(str(N))
        n = N
        s = 0
        while N:
            N, tmp = divmod(N, 10)
            s += tmp ** k
        return s == n