PAT 乙级（Python） #1020 月饼 (25 分)

#1020 月饼 (25 分)

代码展示:

N, D = list(map(eval,input().split()))  # 月饼种类数量, 市场需求
storageLis = list(map(eval,input().split()))  # 月饼库存量
tradeLis = list(map(eval,input().split()))  # 月饼总售价

mooncake_info = []
for i in range(N):
    storage = storageLis[i]
    if storage!=0:
        price = tradeLis[i]/storage
        mooncake_info.append((storage,price))
# print(mooncake_info)
mooncake_info.sort(key=lambda x: x[1],reverse=True)
# print(mooncake_info)

demand_res = D
profit = 0
for storage,price in mooncake_info:
    if demand_res <= 0:
        break
    elif demand_res >= storage:
        demand_res -= storage
        profit += storage * price
    elif demand_res <= storage:
        profit += demand_res * price
        demand_res = 0

print('{0:<.2f}'.format(profit))

测评结果: