[百度笔试复盘] 数字游戏——Python实现

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])

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值