判断是否是水仙花数在Python里怎样输入代码？

n=int(input())

for i in range(n+1):

a=i//100

b=(i-a)//10

c=(i-a-b)//1

if a**3+b**3+c**3==n:

print("有水仙花数")

break

else:

print("没有水仙花数")