有 n 个学生站成一排,每个学生有一个能力值,牛牛想从这 n 个学生中按照顺序选取 k 名学生,要求相邻两个学生的位置编号的差不超过 d,使得这 k 个学生的能力值的乘积最大,你能返回最大的乘积吗?
输入描述:
每个输入包含 1 个测试用例。每个测试数据的第一行包含一个整数 n (1 <= n <= 50),表示学生的个数,接下来的一行,包含 n 个整数,按顺序表示每个学生的能力值 ai(-50 <= ai <= 50)。接下来的一行包含两个整数,k 和 d (1 <= k <= 10, 1 <= d <= 50)。
输出描述:
输出一行表示最大的乘积。
示例1
输入
3
7 4 7
2 50
输出
49
- 假设这道题目只是求最大乘积,那么一维dp就已经够用了,但是这里只要选k个,那么就相当于多了一个维度,就要用二维dp
- 设置dp数组为dpMax[k][i],表示以第i个学生结尾,选了k个学生时候的最大乘积
- dp的时候还有限制条件,因为距离只差不能超过d,因此搜索的时候就要卡上这个限制条件
- 由于数组中有负数,所以还要设置一个最小dp
n = int(input())
line = input()
abilitys = list(map(int, line.strip().split()))
line = input()
k, d = map(int, line.strip().split())
dpMin = [[0]*(n+1) for row in range(k+1)] # 二维list声明方法
dpMax = [[0]*(n+1) for row in range(k+1)]
res = -999999999999
for i in range(1, n+1):
dpMax[1][i] = dpMin[1][i] = abilitys[i-1] # 第i个学生之前选1个,那就初始化他本身。
for m in range(2, k+1): # 遍历k
j = i-1
while j > 0 and i-j <= d: # 遍历这个位置之前到d为止的所有dp数组
dpMax[m][i] = max(dpMax[m][i], max((dpMax[m-1][j] * abilitys[i - 1], dpMin[m-1][j] * abilitys[i - 1])))
dpMin[m][i] = min(dpMin[m][i], min((dpMax[m-1][j] * abilitys[i - 1], dpMin[m-1][j] * abilitys[i - 1])))
j -= 1
res = max(res, dpMax[k][i]) # 看一下第i位置前k各是不是最大值
print(res)