【蓝桥杯每日一练：水仙花】

问题描述

判断给定的三位数是否"水仙花 "数。所谓"水仙花"数是指其值等于它本身每位数字立方和的数。例 153 就是一个 "水仙花" 数。 153=13+53+33

输入格式

一个整数。

输出格式

是水仙花数,输出" YES" ,否则输出" NO" (不包括引号) 。

输入样例

123

输出样例

NO

【提示】

数据规模和约定 ：一个三位的整数,否则输出" NO" 。

代码

import math

while(1):
    n = [int(input())]

    for i in n:
        a = i % 10
        b = i // 10 % 10
        c = i // 100 % 10

        if(math.pow(a,3) + math.pow(b,3) + math.pow(c,3)) == i:
            print('YES')
        else:
            print('NO')