假定我们定义一个数值序列可以被分成两部分,使得这两部分的乘积相等,这样的数列为平衡数,如1236中,1*2*3=6, 为平衡数。现给定一个数列,判断是否为平衡数。
分析:
此题有多种解法,首先最原始的想法是顺次设置分隔符,然后比较分割后的两个子串的乘积,为了提高检查效率,可以首先过滤掉非完全平方数,即数字开方后不为整型的数字,但是这样增加了计算复杂度。
def balance_num_init():
num = sys.stdin.readline().strip()
if len(num)==0:
print "YES"
return
#if isinstance(math.sqrt(num),int):
for i in xrange(1,len(num)):
p1 = num[:i]
p2 = num[i:]
pro1,pro2 = 1,1
for j in range(len(p1)):
pro1 = pro1*int(p1[j])
for j in range(len(p2)):
pro2 = pro2*int(p2[j])
if pro1 ==pro2:
print "YES"
break
else:
pass
if i==len(num)-1:
print "NO"
由于每次计算乘积的时候,都会重复计算之前子串的乘积,所以考虑使用动态规划方法,将计算过的乘积保存起来供下次使用。在这种方法中,我们或许可以先算出分割点之前的所有数的乘积和总乘积,然后用总乘积除以分割点之前的所有数的乘积得到第二部分乘积,判断两者是否相等,但是这样会占用相当大的空间和时间。
所以考虑将数列逆序,得到的数列使用动态规划进行同样的计算,即使用dp[i][j]代表从第i个位置的数字到第j个位置数字的乘积,然后对两个数列分别进行从前向后以及从后向前的对比,若存在相等的比较,则可以判断为平衡数,代码如下。此题应该会有更好的方法,欢迎大家一起讨论。
def dp():
num = sys.stdin.readline().strip()
if len(num)==0:
print "YES"
return
#prod = 1
dp = [[0 for i in range(len(num))] for j in range(len(num))]
rdp = deepcopy(dp)
for i in range(len(num)):
dp[i][i] = int(num[i])
rdp[i][i] = int(num[len(num)-1-i])
#prod = prod*int(num[i])
rnum = num[::-1]
for k in range(len(num)):
for j in xrange(k+1,len(num)):
dp[k][j] = dp[k][j-1]*int(num[j])
for k in range(len(num)):
for j in xrange(k+1,len(num)):
rdp[k][j] = rdp[k][j-1]*int(rnum[j])
#if dp[k][j]*dp[k][j]==prod:
for j in xrange(1,len(num)):
if dp[0][j]==rdp[0][len(num)-2-j] and len(num)-2-j>=0:
print "YES"
return
if j==len(num)-1:
print "NO"