当计算Armstrong数时,我们需要使用一个while循环,以将每个数字的每个digit的立方相加。然而,因为我们需要使用一个for循环来确定数字中的每个digit,所以我们需要把num转换为字符串。这个字符串是可迭代的对象,我们可以使用它来获得每个digit。但是,我们需要把每个字符转换为数字,才能计算其立方。这里的问题是,我们没有把字符转换为数字而直接进行了计算,这就导致了错误的输出。我们可以在代码中添加一个int()函数来解决这个问题。下面是修正后的代码示例:
def is_armstrong(num):
n = len(str(num))
sum = 0
temp = num
while temp > 0:
digit = int(temp % 10)
sum += digit ** n
temp //= 10
if num == sum:
return True
else:
return False
print(is_armstrong(153)) # 输出 True
print(is_armstrong(370)) # 输出 True
print(is_armstrong(371)) # 输出 True
print(is_armstrong(9474)) # 输出 True
print(is_armstrong(9475)) # 输出 False