题目如下:
现给定所有种类月饼的库存量、总售价、以及市场的最大需求量,请你计算可以获得的最大收益是多少。
注意:销售时允许取出一部分库存。样例给出的情形是这样的:假如我们有3种月饼,其库存量分别为18、15、10万吨,总售价分别为75、72、45亿元。如果市场的最大需求量只有20万吨,那么我们最大收益策略应该是卖出全部15万吨第2种月饼、以及5万吨第3种月饼,获得 72 + 45/2 = 94.5(亿元)。
输入格式:
每个输入包含1个测试用例。每个测试用例先给出一个不超过1000的正整数N表示月饼的种类数、以及不超过500(以万吨为单位)的正整数D表示市场最大需求量。随后一行给出N个正数表示每种月饼的库存量(以万吨为单位);最后一行给出N个正数表示每种月饼的总售价(以亿元为单位)。数字间以空格分隔。
输出格式:
对每组测试用例,在一行中输出最大收益,以亿元为单位并精确到小数点后2位。
输入样例:3 20 18 15 10 75 72 45输出样例:
94.50题目解答:
先看代码再做解释:
lst = input().split()
k,m = int(lst[0]),int(lst[1])
num = list(map(float,input().split()))
price = list(map(float,input().split()))
price_s = [[price[i]/num[i],i] for i in range(k)]
price_s.sort(reverse=1)
money = 0
sum_yb = 0
for p,inx in price_s:
sum_yb += num[inx]
if sum_yb <= m:
money += price[inx]
else:
money += p* (num[inx]-(sum_yb-m))
break
print("%.2f"%money)
首先处理输入数据,并得到一个关于每种月饼单价的二维列表 price_s,存放单价及索引(可以使排序后仍保留对应种类)。
money是最终结果的收益,sum_yb用于计算库存加和与总需求m相比。理清思路不难理解。需要注意的一点(其实就是我开始没注意到的一点= =),库存及总售价是正数,需用float类型,否则有一个测试点返回非0,如此积累一些异常处理的经验吧~