Problem C: 水仙花数

Problem Description

所谓水仙花数是指一个三位数，其各位数字的立方和等于该数本身。例如：153 = 1^3 + 5^3 + 3^3，则153是一个水仙花数。要求从键盘上输入一个三位数，判断该数是否是一个水仙花数。

 

 Input Description

 在一行中输入一个三位数x。

 

 Output Description

 如果x是水仙花数，则输出“YES”；否则，输出“NO”。

Sample Input

153

 Sample Output

YES

 Hint

 我的想法：

 我的代码：

data = int(input())


a = int(data % 10)
b = int(data / 10 % 10)
c = int(data / 100)


#encode("是水仙花数", encoding='utf-8')
if a ** 3 + b ** 3 + c ** 3 == data:
    print("YES")
else:
    print("NO")