Python蓝桥杯练习01——水仙花数

题目

水仙花数是指一个N位正整数（N≥3），它的每个位上的数字的N次幂之和等于它本身。 例如：153=1×1×1+5×5×5+3×3×3。

本题要求编写程序,计算所有N位水仙花数。

输入格式:

输入在一行中给出一个正整数N（3≤N≤5）

输出格式:

按递增顺序输出所有N位水仙花数，每个数字占一行。

输入样例:

在这里给出一组输入。例如：

3

输出样例:

在这里给出相应的输出。例如：

153
370
371
407

代码

n=int(input()) #输入[3,5]中的一个数
for i in range(10**(n-1),10**n): #确认范围
    s=0 #水仙花数
    a=i #用于计算
    for j in range(n): #遍历循环
        s+=(a%10)**n #将个位上的数提取并开n次方
        a//=10 #除去个位
        j+=1
    if i == s: #判断
        print(i)

运算结果