p ython求水仙花数pta

在Python中，水仙花数是指一个三位数（或者更长位数，但这里我们通常指三位数），其各位数字立方之和等于该数本身。例如，153是一个水仙花数，因为1^3 + 5^3 + 3^3 = 1 + 125 + 27 = 153。

你可以编写一个简单的函数来检查一个给定的整数是否为水仙花数。下面是一个简单的例子：

def is_lychrel_number(n):
    # 将输入转换为字符串并分割成单个字符
    digits = [int(d) for d in str(n)]
    
    # 对每个位进行立方运算，并相加
    sum_of_cubes = sum(d ** 3 for d in digits)
    
    # 检查循环是否超过10次，如果超过，可能是水仙花数
    for _ in range(10):  # 常规做法最多试10次
        if n == sum_of_cubes:
            return True
        else:
            n = sum_of_cubes
            sum_of_cubes = sum(int(d) ** 3 for d in str(n))
    
    return False

# 测试函数
number = int(input("请输入一个三位数："))
if is_lychrel_number(number):
    print(f"{number} 是一个水仙花数")
else:
    print(f"{number} 不是水仙花数")