leetcode_1134 阿姆斯特朗数

379 阅读1分钟

要求

假设存在一个 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

核心代码

class Solution:
    def isArmstrong(self, n: int) -> bool:
        k = len(str(n))
        _n = n
        s = 0
        while n:
            n,tmp = divmod(n,10)
            s += tmp ** k
        return s == _n

image.png

解题思路:这个题和其他的xx数的题一样,按照规则进行处理即可,比较简单。