电信研发工程师L2考试例题---平衡数

链接： _牛客笔试题_牛客网
来源：牛客网

牛牛在研究他自己独创的平衡数，平衡数的定义是：将一个数分成左右两部分，分别成为两个新的数。
左右部分必须满足以下两点：
1，左边和右边至少存在一位。
2，左边的数每一位相乘如果等于右边的数每一位相乘，则这个数称为平衡数。
例如：1221这个数，分成12和21的话，1*2=2*1，则称1221为平衡数，再例如：1236这个数，可以分成123和1*2*3=6，所以1236也是平衡数。而1234无论怎样分也不满足平衡数。

输入描述:

输入一个正整数（int范围内）。

 

输出描述:

如果该数是平衡数，输出 "YES", 否则输出 "NO"。

示例1

输入

1221
1234

输出

YES
NO

from functools import reduce
str=input()
nums=list(map(int,str))
reuslt='NO'
for i in range(1,len(nums)):
    rl=nums[0:i]
    ll=nums[i:]
    r=reduce(lambda x, y:x*y,rl)
    l=reduce(lambda x, y:x*y,ll)
    if r==l:
        reuslt='YES'
        break
print(reuslt)