python寻找1000以内的阿姆斯特朗数

阿姆斯特朗数定义：如果一个n位正整数等于其各位数字的n次方之和,则称该数为阿姆斯特朗数。 例如1^3 + 5^3 + 3^3 = 153。

1000以内的阿姆斯特朗数： 1, 2, 3, 4, 5, 6, 7, 8, 9, 153, 370, 371, 407。

方法一：计算出位数然后用for循环

核心代码为

length = len(str(num))							# num是输入的数字，类型为int
for ii in range(length):						# length是数字的位数
    num_str = str(num)
    sum_armstrong += int(num_str[ii])**length	  # num_str[]是原数拆分成各个位的数字
    
if sum_armstrong == num:
    return True

完整代码如下：

def is_armstrong(num):
    length = len(str(num))
    sum_armstrong = 0
    for ii in range(length):
        sum_armstrong += int(str(num)[ii])**length

    if sum_armstrong == num:
        return True
    else:
        return False

for num in range(1,1000):
    if is_armstrong(num):
        print(num, end = ' ')

运行效果：

1 2 3 4 5 6 7 8 9 153 370 371 407 
Process finished with exit code 0

方法二：用while循环+判断位数

核心代码：

temp = num									    # 新建一个用于找每一位数字的临时变量
while temp > 0:
    sum_armstrong += (temp%10)**length			  # 取最后一位数字
    temp //= 10									# 除以10，整除复发，相当于直接拿掉最后一位数字
    
if sum_armstrong == num:
    return True

完整代码如下：

def is_armstrong(num):
    temp = num                          
    sum_armstrong = 0
    length = len(str(num))
    while temp > 0:
        sum_armstrong += (temp%10)**length   # 取最后一位数字
        temp //= 10                     
    if sum_armstrong == num:
        return True
    else:
        return False

for num in range(1,1000):
    if is_armstrong(num):
        print(num, end = ' ')

总结：while循环在不确定位数的时候更有效，for循环需要预先计算更多的信息，比如循环次数；而while则不需要，可以创造退出条件，一边计算一边判断。