题目描述
在柠檬水摊上,每一杯柠檬水的售价为 5
美元。
顾客排队购买你的产品,(按账单 bills
支付的顺序)一次购买一杯。
每位顾客只买一杯柠檬水,然后向你付 5
美元、10
美元或 20
美元。你必须给每个顾客正确找零,也就是说净交易是每位顾客向你支付 5
美元。
注意,一开始你手头没有任何零钱。
如果你能给每位顾客正确找零,返回 true
,否则返回 false
。
样例
输入:[5,5,5,10,20]
输出:true
解释:
前 3 位顾客那里,我们按顺序收取 3 张 5 美元的钞票。
第 4 位顾客那里,我们收取一张 10 美元的钞票,并返还 5 美元。
第 5 位顾客那里,我们找还一张 10 美元的钞票和一张 5 美元的钞票。
由于所有客户都得到了正确的找零,所以我们输出 true。
输入:[5,5,10]
输出:true
输入:[10,10]
输出:false
输入:[5,5,10,10,20]
输出:false
解释:
前 2 位顾客那里,我们按顺序收取 2 张 5 美元的钞票。
对于接下来的 2 位顾客,我们收取一张 10 美元的钞票,然后返还 5 美元。
对于最后一位顾客,我们无法退回 15 美元,因为我们现在只有两张 10 美元的钞票。
由于不是每位顾客都得到了正确的找零,所以答案是 false。
注意
0 <= bills.length <= 10000
bills[i]
不是5
就是10
或是20
。
算法
(贪心) $O(n)$
- 开两个变量记录手中 5 元和 10 元的数量。
- 收到 5 元,直接增加一张 5 元;
- 收到 10 元,如果没有 5 元了,则返回 false;
- 收到 20 元,则如果有 10 元的,并且也有至少一张 5 元的,则优先将 10 元配 5 元纸币的找回(因为 5 元的可以更灵活);如果没有 10 元的,但 5 元的有三张,则直接找回三张 5 元的。否则,无法找零,返回 false。
时间复杂度
- 只需遍历一次数组,故时间复杂度为 $O(n)$。
空间复杂度
- 只需要常数个额外的变量,故空间复杂度为 $O(1)$。
C++ 代码
class Solution {
public:
bool lemonadeChange(vector<int>& bills) {
int fives = 0, tens = 0;
for (auto &x : bills) {
if (x == 5)
fives++;
else if (x == 10) {
if (fives == 0)
return false;
tens++;
fives--;
}
else {
if (tens >= 1 && fives >= 1) {
tens--;
fives--;
}
else if (tens == 0 && fives >= 3) {
fives -= 3;
}
else
return false;
}
}
return true;
}
};