2020.3.29日 测试开发岗位 笔试编程题复盘
数字游戏
- 题目描述
小W发明了一个游戏,他在黑板上写出了一行数字a1,a2,a3,……,an,然后给你M个回合的机会,每回合你可以从中选择一个数字擦去它,接着剩下来的每个数字ai都要递减一个值bi,即a1减掉b1,a2减掉b2,….。如此重复M个回合,所有你擦去的数字之和就是你所得的分数。
小W和他的好朋友小Y玩了这个游戏,可是他发现,对于每个给出的an和bn序列,小Y的得分总比他高,所以他就很不服气。于是他想让你帮他算算,对于每个an和bn序列,可以得到的最大得分是多少。这样他就知道有没有可能超过小Y的得分。
- 输入
输入文件的第一行是一个整数n(1<=n<=2000),表示数字个数;第二行一个整数m(1<=m<=n),表示回合数,接下来一行有n个不超过10000的正整数,a1,a2,a3,……,an表示原始序列,最后一行有n个不超过500的正整数,b1,b2,b3,……,bn,表示每回合每个数字递减的值。 - 输出
输出文件只有一个整数,表示最大的可能得分。 - 样例输入
3
3
10 20 30
4 5 6 - 样例输出
47
解题思路
这是一道应用动态规划解题思路的题。
用f[i][j]表示a中前i个数字中选j回合所能取到的最大值。
所以f[i][j]有两种状态转移过来
(1)选择了当前的a[i]个点:f[i][[j]=f[i-1][j-1]+a[i]-b[i]*(j-1)
(2)没有选择当前这个点:f[i][j]=f[i-1][j]
代码实现
#!/usr/bin/env python
# -*- coding:utf-8 -*-
n = int(input())
m = int(input())
a = list(map(int, input().split()))
b = list(map(int, input().split()))
# 先将原数组按照b数据大小进行排序
for i in range(m):
for j in range(i + 1, m):
if b[i] < b[j]:
b[i], b[j] = b[j], b[i]
a[i], a[j] = a[j], a[i]
# 初始化数组
f = [[0] * (m + 1) for _ in range(n + 1)]
# 开始进行dp
for i in range(1,n+1):
#边界值
f[i-1][0]=0
for j in range(1, m + 1):
#状态转移式子
f[i][j] = max(f[i - 1][j], f[i - 1][j - 1] + a[i - 1] - b[i - 1] * (j - 1))
print(f[n][m])