leetcode_860. 檸檬水找零

阿新 • • 發佈：2020-12-11

技術標籤：leetcode Python leetcode 演算法貪心演算法 python

一、題目內容

在檸檬水攤上，每一杯檸檬水的售價為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

二、解題思路

1.這道題需要知道的是10元需要找5元，20元需要找1個10元和1和5元，或者3個5元。

2.那麼10元找5元，5元的個數不能為0；

3.20元找15元：

(1)如果5元少於3個，則只能看看能不能找1個10元和1個5元；

(2)如果5元足夠的話，既可以找3個5元也可以找1個10元和1個5元。

三、程式碼

class Solution:
    def lemonadeChange(self, bills: list) -> bool:
        bills_dict = {
            '5': 0,
            '10': 0,
            '20': 0,
        }
        for num in bills:
            if num == 5:
                bills_dict['5'] += 1
            elif num == 10:
                if bills_dict['5'] < 1:
                    return False
                bills_dict['5'] -= 1
                bills_dict['10'] += 1
            elif num == 20:
                if bills_dict['5'] < 3:
                    if bills_dict['10'] < 1 or bills_dict['5'] < 1:
                        return False
                    bills_dict['10'] -= 1
                    bills_dict['5'] -= 1
                    bills_dict['20'] += 1
                else:
                    if bills_dict['10'] < 1:
                        bills_dict['5'] -= 3
                        bills_dict['20'] += 1
                    else:
                        bills_dict['10'] -= 1
                        bills_dict['5'] -= 1
                        bills_dict['20'] += 1
        return True


if __name__ == '__main__':
    bills = [5,5,5,10,20]
    s = Solution()
    ans = s.lemonadeChange(bills)
    print(ans)