【浙大版《Python 程序设计》题目集（解）】第4章-17 水仙花数（20 分）

水仙花数是指一个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（3≤N≤5）
n = int(input())


# 计算水仙花数并输出符合条件的水仙花数
lst = list()  # 保存各个位上的数

for item in range(10 ** (n - 1), 10 ** n + 1):
    
    sums = 0  # 保存各个位上的n次方总和
    flag = item  # 用于判断
    
    for i in range(n):  # 取出各个位上的数
        lst.append(item % 10)  
        item //= 10 
    
    for j in range(n):  # 计算各个位上的n次方总和
        sums += lst[j] ** n
    
    if sums == flag:  # 判断是否是水仙花数
        print(flag)
    
    lst.clear()  # 列表清空

##########################################################################


# 方案2

# 输入在一行中给出一个正整数N（3≤N≤5）
n = int(input())


# 计算水仙花数并输出符合条件的水仙花数
for i in range(10 ** (n - 1), 10 ** n):

    sums = 0
    flag = i
    
    while i >= 1:
        sums += (i % 10) ** n
        i //= 10

    if sums == flag:
        print("{}".format(flag))

如有更好的建议请评论！！！