题目描述:
王强决定把年终奖用于购物,他把想买的物品分为两类:主件与附件,附件是从属于某个主件的,下表就是一些主件与附件的例子:
如果要买归类为附件的物品,必须先买该附件所属的主件,且每件物品只能购买一次。每个主件可以有 0 个、1 个或 2 个附件。附件不再有从属于自己的附件。
王强查到了每件物品的价格(都是 10 元的整数倍),而他只有 N 元的预算。除此之外,他给每件物品规定了一个重要度,用整数 1 ~ 5 表示。他希望在花费不超过 N 元的前提下,使自己的满意度达到最大。
其中,满意度是指所购买的每件物品的价格与重要度的乘积的总和,假设设第ii件物品的价格为v[i]v[i],重要度为w[i]w[i],共选中了kk件物品,编号依次为j1,j2,...,jkj1,j2,...,jk,则满意度为:v[j1]∗w[j1]+v[j2]∗w[j2]+…+v[jk]∗w[jk]v[j1]∗w[j1]+v[j2]∗w[j2]+…+v[jk]∗w[jk]。(其中 * 为乘号)
请你帮助王强计算可获得的最大的满意度。
输入描述:
输入的第 1 行,为两个正整数N,m,用一个空格隔开:(其中 N ( N<32000 )表示总钱数, m (m <60 )为可购买的物品的个数。)
从第 2 行到第 m+1 行,第 j 行给出了编号为 j-1 的物品的基本数据,每行有 3 个非负整数 v p q(其中 v 表示该物品的价格( v<10000 ), p 表示该物品的重要度(1-5), q 表示该物品是主件还是附件。如果 q=0 ,表示该物品为主件,如果 q>0 ,表示该物品为附件, q 是所属主件的编号)
输出描述:
输出一个正整数,为张强可以获得的最大的满意度。
def max_satisfaction(n,m,items):
# 调整items列表从1开始编号
items=[0]+items
# 更新动态规划数组dp
dp=[0]*(n+1)
for i in range(1,m+1):
if items[i][2]==0:
for j in range(n,items[i][0]-1,-1):
dp[j]=max(dp[j],dp[j-items[i][0]]+items[i][0]*items[i][1])
return dp[n]
# 输入
n,m=map(int,input().split())
items=[]
for _ in range(m):
v,p,q=map(int,input().split())
items.append([v,p,q])
print(max_satisfaction(n,m,items))
上述代码是一个由动态规划迭代来实现的,首先,定义了一个名为max_satisfaction的函数,接受三个参数:总钱数n,购买数量m,以及一个表示物品的价格、重要度和主件/附件的列表items。然后,使用动态规划算法获得最大满意度。接下来,检查该物品是否为主件(items[i][2]是否为0),如果是主件,代码会继续遍历所有可能的钱数j(从n开始,到物品价格items[i][0]-1)。对于每个钱数j,代码计算购买物品i和不购买物品i两种情况下的总满意度,然后选择满意度较高的一种。动态规划数组dp的每个元素dp[j]将被更新为在钱数为j的情况下可以获得的最大满意度。最后,dp[n]将保存总钱数为n时获得的最大满意度。所以,最后返回dp[n],即函数需要输出的结果。