Day 65: 柠檬水找零

在柠檬水摊上,每一杯柠檬水的售价为 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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/lemonade-change
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

def lemonadeChange(bills):
    mybox = {5:0,10:0} #用于保存我钱箱里的钱的数量,5,10元的钞票各有几张
    for bill in bills:
        if bill == 5: #收到5元
            mybox[5] += 1
        elif bill == 10: #收到10元
            if mybox[5] >=1: #有零钱可以找,收进10元1张,找出5元1张
                mybox[10] += 1
                mybox[5] -= 1
            else:
                return False
        else: #还有一种可能是收到20元
            if mybox[10] >=1 and mybox[5] >=1: #找出10元1张,5元1张
                mybox[10] -= 1
                mybox[5] -= 1
            elif mybox[5] >=3: #有3张5元,也可以发出
                mybox[5] -= 3
            else:
                return False
        return True


def test_lemonadeChange():
    assert lemonadeChange([5,5,5,10,20]) == True
    assert lemonadeChange([5,5,10]) == True
    assert lemonadeChange([10,10]) == False
    assert lemonadeChange([5,5,10,10,20]) == False

你可能感兴趣的:(Day 65: 柠檬水找零)