（python）阶乘和数

一个正整数如果等于组成它的各位数字的阶乘之和，则该正整数称为阶乘和数。例如正整数145，1!+4!+5!等于145，因此145就是一个阶乘和数。输入一个正整数，计算它的各位数字的阶乘之和，判断它是否是一个阶乘和数。当输入的数字为阶乘和数时，输出“YES”,否则输出“NO”。

解决代码如下：

n=int(input())
k=str(n)
f=0
def jiecheng(e):
    s=1
    for i in range(1,e+1):
        s=s*i
    return s
for i in k:
    j=int(i)
    p=jiecheng(j)
    f=f+p
if f==n:
    print('YES')
else:
    print('NO')