PAT-1020：月餅 Python3（25）（知識點：貪心算法）

1020 月餅 (25分)

題目：

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

注意：銷售時允許取出一部分庫存。樣例給出的情形是這樣的：假如我們有 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

---

題解

方法一：

n, c = list(map(int, input().split()))  
num= list(map(float, input().split())) 
money= list(map(float, input().split())) 
rate = {i: money[i]/num[i] for i in range(n)} 
order = sorted(rate, key=lambda i:rate[i], reverse=True) 
summoney= 0 
for i in order:   
    if c>= money[i]:
        summoney += money[i]
        c-= num[i]
    else:
        summoney+= c* rate[i]
        break
print('%.2f' % summoney)

方法二：

n,c=map(int,input().split())
num=list(map(float ,input().split() ))
money=list(map(float,input().split()))
s=[float(money[i]/num[i]) for i in range(n)]
summoney=0
while c > 0:
    maxi = int(s.index(max(s)))
    if max(s)==-1:  # 這裏需要判斷一下是否所有月餅都賣完了
        break
    if c >= num[maxi]:
        summoney+=money[maxi]
        s[maxi]=-1
        c-=num[maxi]
    else:
        summoney+=s[maxi]*c
        break
print("%.2f" % summoney)

在方法二中，如果沒有 if max(s)==-1: break這一條判斷退出語句，最後結果回事部分正確！

原因是：當遇到市場需求很大時，所有月餅都賣完了，這樣的話，因爲c一直大於0成立，我們就要自己主動跳出循環