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 

问题分析

此题用贪心算法。遍历给定数组bills，如果钱数是5，那么++umap[5]；如果钱数是10，那么++umap[10]，然后判断哈希map中5的映射值，如果大于0说明有找零的，那么--umap[5]，否则返回false；如果钱数是20，那么先判断是否有一张5元的和一张10元的，有则用它们找零，如果不满足该条件，那么判断如果5元的有至少3张，说明可以用它们来找零，那么将umap[5]减3，否则如果上面的条件都不满足，则返回false。当bills数组遍历结束后，返回true。

代码实现

class Solution {
public:
    bool lemonadeChange(vector<int>& bills) {
        unordered_map<int, int> umap;
        for(int i : bills){
            if(i == 5)
                ++umap[5];
            else if(i == 10){
                ++umap[10];
                if(umap[5])
                    --umap[5];
                else
                    return false;
            }
            else{
                if(umap[10] && umap[5]){
                    --umap[10];
                    --umap[5];
                }
                else if(umap[5] > 2)
                    umap[5] -= 3;
                else
                    return false;
            }
        }
        return true;
    }
};