水仙花数

n = int(input())
bai = n // 100
shi = n % 100 // 10
ge = n % 10
if n == bai ** 3 + shi ** 3 + ge ** 3:
print('YES')
else:
print('NO')
这是我写的Python答案