ALGO-17 VIP试题 乘积最大 动态规划 python

问题描述

　　今年是国际数学联盟确定的“2000——世界数学年”，又恰逢我国著名数学家华罗庚先生诞辰90周年。在华罗庚先生的家乡江苏金坛，组织了一场别开生面的数学智力竞赛的活动，你的一个好朋友XZ也有幸得以参加。活动中，主持人给所有参加活动的选手出了这样一道题目：

　　设有一个长度为N的数字串，要求选手使用K个乘号将它分成K+1个部分，找出一种分法，使得这K+1个部分的乘积能够为最大。

　　同时，为了帮助选手能够正确理解题意，主持人还举了如下的一个例子：

　　有一个数字串：312， 当N=3，K=1时会有以下两种分法：

　　3*12=36
　　31*2=62

　　这时，符合题目要求的结果是：31*2=62

　　现在，请你帮助你的好朋友XZ设计一个程序，求得正确的答案。

输入格式

　　程序的输入共有两行：
　　第一行共有2个自然数N，K（6≤N≤40，1≤K≤6）
　　第二行是一个长度为N的数字串。


输出格式

　　输出所求得的最大乘积（一个自然数）。

　　样例输入

　　4 2
　　1231

样例输出

62

思路：动态规划思想，dp[a][b]表示从0到a的一列数字有b个乘号时的最大结果。定义函数count(list)计算list数组的数字组成的值为多少。比如count([1,3,2])返回值132。

状态转移方程：

dp[a][b] = max(dp[a-i][b-1]*count(list[a-i+1:i+1]))，

当b = 0 时，dp[a][0] = count(list[0:a+1])

N,K = map(int,input().split())
Nums = list(input())
nums = []
for i in Nums:
    nums.append(int(i))
# print(len(nums))
dp = [[1 for i in range(K+1)]for j in range(N)]
def count(list):
    cif = 1
    ans = 0
    while(len(list)>0):
        ans += cif*list.pop(-1)
        cif *=10
    return ans
# print(count([1,2,3]))

for i in range(N):
    dp[i][0] = count(nums[0:i+1])
for i in range(0,N):
    for j in range(1,min(i+1,K+1)):
        for p in range(1,i-j+2):
            dp[i][j] = max(dp[i][j],dp[i-p][j-1]*count(nums[i-p+1:i+1]))
print(dp[N-1][K])