如何使用python AK这道题

我们可以用动态规划来解，首先定义一个二维数组$dp[i][j]$，表示在前$i$个字符中插入$j$个乘号所能得到的最大乘积。
对于每个$dp[i][j]$，我们可以通过枚举最后一个乘号的位置$k$，将问题分解为前$k$个字符插入$j-1$个乘号和后$i-k$个字符的乘积。状态转移方程为：
$dp[i][j] = max(dp[i][j], dp[k][j-1] * int(num[k:i]))$
其中$num$是输入的数字串，$int(num[k:i])$表示从第$k$个字符到第$i$个字符组成的子串的数值。
$dp[i][0]$表示前i个字符不插入任何乘号的情况，即直接将前$i$个字符作为一个整体。
题解：

def max_product(N, K, num_str): 
    # 初始化dp数组 
    dp = [[0] * (K + 1) for _ in range(N + 1)] 
     
    # 初始化dp[i][0]，即不插入任何乘号的情况 
    for i in range(1, N + 1): 
        dp[i][0] = int(num_str[:i]) 
     
    # 动态规划求解 
    for j in range(1, K + 1): 
        for i in range(j + 1, N + 1): 
            for k in range(j, i): 
                dp[i][j] = max(dp[i][j], dp[k][j - 1] * int(num_str[k:i])) 
     
    return dp[N][K] 
 
# 输入 
N, K = map(int, input().split()) 
num_str = input().strip() 
 
# 输出结果 
print(max_product(N, K, num_str)) 

对于输入1231，使用2个乘号分成3个部分，最大乘积为12 * 3 * 1 = 36。