题目描述:
解题思路:
分析:
先看题,从题目描述和下面的提示,大家都可以知道。这个bills数组中的值,只有三种可能——5、10、20。一个顾客只买一杯柠檬水,也就是每个订单只消费5美元。
代码思路:
这道题非常简单,直接遍历bills数组,对每一个顾客的账单进行处理。
(1)如果一个顾客支付了5美元,那么不需要找零。我们直接将5美元收下即可。
(2)如果一个顾客支付了10美元,那么我们只有一种找零方式,就是支付顾客5美元。
(3)如果一个顾客支付了20美元,我们有两种找零方式:(1)支付顾客一张10美元+一张5美元(2)支付顾客三张5美元
疑问:哪里用到贪心算法了?
在第三种情况下,我们有两种找零的方式。我们优先使用第一种找零方式:10美元+5美元。
我们可以思考一下:
5美元既可以对情况(2)找零,也可以对情况(3)找零。而10美元只能对情况(3)找零。所以为了给更多的顾客找零,我们优先使用10美元+5美元的组合对情况(3)找零。
代码和注释:
class Solution:
def lemonadeChange(self, bills: List[int]) -> bool:
# 5美元的张数
five_count = 0
# 10美元的张数
ten_count = 0
# 遍历账单
for i in range(len(bills)):
# 顾客给的是5美元
if bills[i] == 5:
five_count += 1
# 顾客给的是10美元
if bills[i] == 10:
if five_count > 0:
ten_count += 1
five_count -= 1
else:
return False
# 顾客给的是20美元
if bills[i] == 20:
# 优先给5+10的组合是贪心策略的体现
if five_count > 0 and ten_count > 0:
five_count -= 1
ten_count -= 1
elif five_count >= 3:
five_count -= 3
else:
return False
# 都成功找零了
return True