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