題目:
檸檬水找零: 在檸檬水攤上,每一杯檸檬水的售價為 5 美元。 顧客排隊購買你的產品,(按賬單 bills 支付的順序)一次購買一杯。 每位顧客只買一杯檸檬水,然后向你付 5 美元、10 美元或 20 美元。你必須給每個顧客正確找零,也就是說凈交易是每位顧客向你支付 5 美元。
注意,一開始你手頭沒有任何零錢。
如果你能給每位顧客正確找零,返回 true ,否則返回 false 。
示例 1:
輸入:[5,5,5,10,20]
輸出:true
解釋:
前 3 位顧客那里,我們按順序收取 3 張 5 美元的鈔票。
第 4 位顧客那里,我們收取一張 10 美元的鈔票,並返還 5 美元。
第 5 位顧客那里,我們找還一張 10 美元的鈔票和一張 5 美元的鈔票。
由於所有客戶都得到了正確的找零,所以我們輸出 true。
示例 2:
輸入:[5,5,10]
輸出:true
示例 3:
輸入:[10,10]
輸出:false
示例 4:
輸入:[5,5,10,10,20]
輸出:false
解釋:
前 2 位顧客那里,我們按順序收取 2 張 5 美元的鈔票。
對於接下來的 2 位顧客,我們收取一張 10 美元的鈔票,然后返還 5 美元。
對於最后一位顧客,我們無法退回 15 美元,因為我們現在只有兩張 10 美元的鈔票。
由於不是每位顧客都得到了正確的找零,所以答案是 false。
提示:
0 <= bills.length <= 10000
bills[i] 不是 5 就是 10 或是 20
思路:
從總體夠不夠找和細節是否可以找入手。
程序:
class Solution: def lemonadeChange(self, bills: List[int]) -> bool: if not bills: return True if bills[0] > 5: return False auxiliary = [0] * (len(bills) + 1) num_5 = 0 num_10 = 0 for index in range(len(bills)): if bills[index] == 5: auxiliary[index + 1] = auxiliary[index] + bills[index] num_5 += 1 elif bills[index] == 10: auxiliary[index + 1] = auxiliary[index] + (bills[index] - 5) num_10 += 1 num_5 -= 1 else: auxiliary[index + 1] = auxiliary[index] + (bills[index] - 5) if num_10 == 0: num_5 -= 3 else: num_5 -= 1 num_10 -= 1 if num_5 < 0 or num_10 < 0: return False for index2 in range(1, len(auxiliary)): if auxiliary[index2] < 0: return False return True