PAT (Basic Level) Practice （中文）1020 月饼 python

出处：
https://pintia.cn/problem-sets/994805260223102976/problems/994805301562163200

本题的解法就是先买最贵的，然后依次往下买，直到达到最大需求量。
代码：

n = list(map(int,input().split()))
num = input().split()
price = input().split()

N = n[0]    #N为月饼种类数
D = n[1]    #D为市场最大需求量
res = 0     #res为最后返回结果

cake_price = {i: float(price[i]) / float(num[i]) for i in range(N)}     #cake_price为月饼单价（每万顿）
cake_price_sorted = sorted(cake_price,key=lambda x:cake_price[x],reverse = True)

for i in cake_price_sorted:     #如果剩下的需求量比现在最贵的月饼总数多，直接加上所有现在最贵的月饼
    if D >= float(num[i]):
        res += int(price[i])
        D -= float(num[i])
    else:
        res += D * cake_price[i]    #如果不如，用剩下需求乘以目前最贵单价
        break

print('%.2f' % res)