leetcode 860.柠檬水找零

leetcode 860.柠檬水找零

题干

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

题解

模拟，考虑不同的找钱策略即可，唯一要注意的点是20块有两种找零的方法：5+10,5+5+5。
用哈希表储存数据，因为20块钱不能用来找零所以20块的数据可以不记录

class Solution {
public:
    bool lemonadeChange(vector<int>& bills) {
        unordered_map<int,int> dollarInHand;
        int n = bills.size();
        if(n == 0){
            return true;
        }else if(bills[0] != 5){
            return false;
        }
        //确保开始遍历时第一张钞票一定是五块
        for(int i = 0 ; i < n ; ++i){
            if(bills[i] == 5){
                dollarInHand[5]++;
            }else if(bills[i] == 10){
                if(dollarInHand[5] > 0){
                    dollarInHand[5]--;
                    dollarInHand[10]++;
                }else{
                    return false;
                }
            }else if(bills[i] == 20){
                if(dollarInHand[10] > 0 && dollarInHand[5] > 0){
                    dollarInHand[5]--;
                    dollarInHand[10]--;
                }else if(dollarInHand[5] >= 3){
                    dollarInHand[5] -= 3;
                }else{
                    return false;
                }
            }
        }
        return true;
    }
};

不用unordered_set还可以开个数组，把面值映射到下标上，开两位即可，思路是一样的

class Solution {
public:
    bool lemonadeChange(vector<int>& bills) {
        vector<int> dollarInHand(2,0);
        int n = bills.size();
        if(n == 0){
            return true;
        }else if(bills[0] != 5){
            return false;
        }
        for(int i = 0 ; i < n ; ++i){
            int index = bills[i] / 5 - 1;
            if(bills[i] == 5){
                dollarInHand[index]++;
            }else if(bills[i] == 10){
                if(dollarInHand[0]){
                    dollarInHand[0]--;
                    dollarInHand[index]++;
                }else{
                    return false;
                }
            }else if(bills[i] == 20){
                if(dollarInHand[1] && dollarInHand[0]){
                    dollarInHand[0]--;
                    dollarInHand[1]--;
                }else if(dollarInHand[0] >= 3){
                    dollarInHand[0] -= 3;
                }else{
                    return false;
                }
            }
        }
        return true;
    }
};