阿姆斯特朗数定义:如果一个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则不需要,可以创造退出条件,一边计算一边判断。