【PAT乙級】月餅

題目描述：

月餅是中國人在中秋佳節時吃的一種傳統食品，不同地區有許多不同風味的月餅。現給定所有種類月餅的庫存量、總售價、以及市場的最大需求量，請你計算可以獲得的最大收益是多少。

注意：銷售時允許取出一部分庫存。樣例給出的情形是這樣的：假如我們有 3 種月餅，其庫存量分別爲 18、15、10 萬噸，總售價分別爲 75、72、45 億元。如果市場的最大需求量只有 20 萬噸，那麼我們最大收益策略應該是賣出全部 15 萬噸第 2 種月餅、以及 5 萬噸第 3 種月餅，獲得 72 + 45/2 = 94.5（億元）。

輸入格式：

每個輸入包含一個測試用例。每個測試用例先給出一個不超過 1000 的正整數 N 表示月餅的種類數、以及不超過 500（以萬噸爲單位）的正整數 D 表示市場最大需求量。隨後一行給出 N 個正數表示每種月餅的庫存量（以萬噸爲單位）；最後一行給出 N 個正數表示每種月餅的總售價（以億元爲單位）。數字間以空格分隔。

輸出格式：

對每組測試用例，在一行中輸出最大收益，以億元爲單位並精確到小數點後 2 位。

輸入樣例：

3 20
18 15 10
75 72 45

輸出樣例：

94.50

代碼：

def main():
    n, d = map(int, input().split())
    if n<=0 or d<=0:
        return
    a = [int(i) for i in input().split()]
    b = [int(i) for i in input().split()]
    c = {}
    for i in range(n):
        if a[i]!=0:
            c[i] = b[i]/a[i]
        else:
            c[i] = b[i]
    c=sorted(c.items(), key=lambda c: c[1], reverse=True)
    profit=0.0
    for i in range(n):
        if d==0:
            break
        if a[c[i][0]]<d:
            d=d-a[c[i][0]]
            profit=profit+b[c[i][0]]
        elif a[c[i][0]]>=d:
            profit=profit+d*c[i][1]
            d=d-d
    print('%.2f' % profit)
 
 
if '__name__'==main():
    main()