6-2 编写函数判断水仙花数

6-2 编写函数判断水仙花数

分数 10
作者 王祥荣
单位 宁波财经学院
给定一个三位数n，要求编写函数isflower(n)判断n是否为水仙花数,如果是，则返回True，否则返回False。所谓"水仙花数"是指一个3位数，其各位数字立方和等于该数本身。例如，153是一个水仙花数，因为153等于1的立方加上5的立方加上3的立方之和。

函数接口定义：

在这里描述函数接口。例如：

isflower (n)

其中 n 是传入的参数，表示需要判断的三位数。 n 的值的范围为[100,999]。

裁判测试程序样例：

/* 请在这里填写答案 */

n = eval(input())
if isflower(n)==True:
    print('Yes')
else:
    print('No') 

输入样例：

在这里给出一组输入。例如：

371

输出样例：

在这里给出相应的输出。例如：

Yes

代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

解题思路

本题比较经典，先求一个三位数的每一位数字，然后判断即可。

def isflower(num):
    import math
    bai = num // 100
    ge = num % 10
    shi = (num // 10) % 10
    sum = math.pow(ge, 3) + math.pow(shi, 3) + math.pow(bai, 3)
    return True if sum == num else False