LeetCode 860. 柠檬水找零 Python3

在柠檬水摊上，每一杯柠檬水的售价为 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):
        """
        :type bills: List[int]
        :rtype: bool
        """
        count_5 = 0
        count_10 = 0
        flag = 0

        for i in range(0, len(bills)):
            if bills[i] - 5 == 0:
                count_5 += 1
                flag += 1
            elif bills[i] - 5 == 5:
                if count_5 > 0:
                    count_5 -= 1
                    count_10 += 1
                    flag += 1
                else:
                    return False
                    break

            elif bills[i] - 5 == 15:
                if count_10 > 0 and count_5 > 0:
                    count_10 -= 1
                    count_5 -= 1
                    flag += 1
                elif count_5 >= 3 and count_10 == 0:
                    count_5 -= 3
                    flag += 1
                else:
                    return False
                    break

        if flag == len(bills):
            return True

简单题。。我用了这种暴力解法写的。。其实问题唯一的难点就是，怎么在处理找15块钱的时候的最优办法，因为十五块可能是5+5+5或者5+10，其实仔细想一想吧，能用十块就用十块一定就是最优的。