【Python】一段简单的代码实现寻找水仙花数

不多废话直接上代码：

def find_narcissistic_numbers(n):
    narcissistic_numbers = []
    for num in range(10**(n-1), 10**n):
        temp = num
        sum = 0
        while temp > 0:
            digit = temp % 10
            sum += digit ** n
            temp //= 10
        if sum == num:
            narcissistic_numbers.append(num)
    return narcissistic_numbers

while True:
    n=input("请输入您要寻找的水仙花数位数：")
    if n=="exit":
        break;
    elif n.isdigit():
        n=int(n);
        narcissistic_numbers = find_narcissistic_numbers(n)
        print(str(n) + "位水仙花数:", narcissistic_numbers)
    else:
        print("输入有误，请重新输入")

上述代码实现了一个功能，该功能是通过输入位数 n，找出对应位数的水仙花数并进行打印输出。

首先定义了一个函数 find_narcissistic_numbers(n)，它接受一个参数 n，表示要找出的水仙花数的位数。函数内部通过一个循环遍历从 10 的 (n-1) 次方到 10 的 n 次方之间的所有数字，即范围为 [10^(n-1), 10^n)。对于每个数字 num，在一个内部循环中，将 num 的每一位的数字取出，计算每位数字的 n 次方并累加到 sum 变量中。最后，如果 sum 等于 num，则说明 num 是水仙花数，将其添加到 narcissistic_numbers 列表中。函数返回找到的水仙花数列表。

接下来是一个无限循环，不断询问用户输入要寻找的水仙花数的位数。用户可以输入一个位数 n 或输入 "exit" 来退出程序。程序通过判断用户的输入来执行不同的逻辑：

1.如果用户输入 "exit"，则跳出循环，程序结束。
2.如果用户输入的是一个数字，通过调用 isdigit() 方法判断输入是否为纯数字，然后将其转换为整数类型，并调用 find_narcissistic_numbers(n) 函数找到对应位数的水仙花数，并打印输出。
3.如果用户输入的既不是 "exit" 也不是数字，则提示输入有误，要求重新输入。

通过以上代码，用户可以反复输入不同位数的水仙花数进行查找，并得到相应的结果输出。如果输入位数无效或不是数字，会进行相应的错误处理。

具体效果图：

【Python】一段简单的代码实现寻找水仙花数

猜你喜欢