HJ16 购物单 备忘记录

输入描述：

输入的第 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 是所属主件的编号）

输出描述：

输出一个正整数，为张强可以获得的最大的满意度。

输入：1000 5

800 2 0

400 5 1

300 5 1

400 3 0

500 2 0

输出：2200

输入：50 5

20 3 5

20 3 5

10 3 0

10 2 0

10 1 0

输出：130

说明：

由第1行可知总钱数N为50以及希望购买的物品个数m为5；

第2和第3行的q为5，说明它们都是编号为5的物品的附件；

第4~6行的q都为0，说明它们都是主件，它们的编号依次为3~5；

所以物品的价格与重要度乘积的总和的最大值为10*1+20*3+20*3=130

from collections import defaultdict

balance, k = map(int, input().split())
balance //= 10

main_goods = defaultdict(lambda: [0, 0, 0]) # 商品价格
main_value = defaultdict(lambda: [0, 0, 0]) # 商品价值

for i in range(k):
    v, p, q = map(int, input().split())
    v //= 10

    if q == 0: # 添加主要商品
        main_goods[i + 1][0] = v    #添加商品价格
        main_value[i + 1][0] = v * p    # 添加商品价值

    elif main_goods[q][1]: # 添加从物品
        main_goods[q][2] = v
        main_value[q][2] = v * p

    else:
        main_goods[q][1] = v
        main_value[q][1] = v * p


dp = [0] * (balance + 1)

for i ,v in  main_goods.items():
    for j in range(balance, v[0] -1, -1):
        p1, p2, p3 = v
        v1, v2, v3 = main_value[i]

        dp[j] = max(dp[j], dp[j- p1] + v1)
        dp[j] = max(dp[j], dp[j - p1 - p2] + v1 + v2) if j>=p1 + p2 else dp[j]
        dp[j] = max(dp[j], dp[j - p1 - p3] + v1 +v3) if j >= p1 + p3 else dp[j]
        dp[j] = max([dp[j], dp[j - p1 -p2 - p3 ] + v1 + v2 + v3]) if j >= p1 + p2 + p3 else dp[j]

print(dp[balance] * 10)