PAT乙级(1010)贪心算法-数月饼真题 (python版本)

题目描述

月饼是中国人在中秋佳节时吃的一种传统食品，不同地区有许多不同风味的月饼。现给定所有种类月饼的库存量、总售价、以及市场的最大需
求量，请你计算可以获得的最大收益是多少。

注意：销售时允许取出一部分库存。样例给出的情形是这样的：假如我们有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

代码实现：

#贪心算法求解该问题
def main():
    #月饼种类，市场容纳量
    kinds,total = input().split(" ")
    stock = input().split(" ")
    price = input().split(" ")
    list_c = []
    list_z = []
    for i in range(int(kinds)):
        avg = round(int(price[i])/int(stock[i]),1)
        list_c.append(avg)
        list_z.append([int(stock[i]),avg,0])
    list_c.sort(reverse=True)
    #两个列表保存贪心最大策略取值，存取量
    sum =[0,0]# 用于存放取走的总重量，第一个参数是取走的重量，第二个是超出前的备份
    cost =0
    ji = 0
 
    for i in range(len(list_c)):
        for k in range(len(list_z)):
            if ji == 0:#是否超出市场最大承受量的标记，未超出为0
                if (list_c[i] == list_z[k][1]) and (list_z[k][2]==0):
                    sum[1] = sum[0]#备份
                    sum[0] = sum[0] + list_z[k][0]#取走的重量
                    v = list_z[k][0]#取走的重量
                    if sum[0] > int(total):#如果所有取走的重量超出市场的重量，就依次减少一单元的重量
                        ji = 1#超出为1
                        t= list_z[k][0]
                        while True:#依次减去单位1的重量
                            z = sum[1] + t#使用备份进行判断，此时取走的数量已经大于最大承受量了
                            if z <= int(total):
                                break
                            t = t-1
                        v=t#等于最大承受量时，价值较大的一件物品应取走的数量
                        sum[0]=sum[1]#从备份恢复
                        sum[0] = sum[0] + t#此时为真正的取走数量
                    cost = list_c[i]*v + cost#总价值
                    list_z[k][2] = 1#取走的标记
    print ("{:.2f}".format(cost))
if __name__ =="__main__":
    main()

只通过部分测试样例