leetcode刷题：贪心算法10（柠檬水找零）

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

哈希表做，如果第一次不是来五元，直接错。如果是，就哈希表添加元素。每次找零对应位置相减即可，尽量先减10，后减5.

package com.programmercarl.greed;

import java.util.HashMap;

/**
 * @ClassName LemonadeChange
 * @Descriotion TODO
 * @Author nitaotao
 * @Date 2022/7/16 12:20
 * @Version 1.0
 * https://leetcode.cn/problems/lemonade-change/
 * 860. 柠檬水找零
 **/
public class LemonadeChange {
    public boolean lemonadeChange(int[] bills) {
        if (bills[0] != 5) {
            return false;
        }
        HashMap<Integer, Integer> retains = new HashMap<Integer, Integer>();
        for (int i = 0; i < bills.length; i++) {
            // bills只能是 5 10 20
            switch (bills[i]) {
                case 5:
                    // 获取 5 元的数量，在原有的基础上+1
                    retains.put(5, retains.getOrDefault(5, 0) + 1);
                    break;
                case 10:
                    // 10元+1，5元-1
                    Integer fiveNums = retains.getOrDefault(5, 0);
                    if (fiveNums == 0) {
                        return false;
                    } else {
                        fiveNums--;
                        // 获取 5 元的数量，在原有的基础上+1
                        retains.put(5, fiveNums);
                        retains.put(10, retains.getOrDefault(10, 0) + 1);
                    }
                    break;
                case 20:
                    // 20元+1
                    retains.put(20, retains.getOrDefault(20, 0) + 1);

                    // 10元-1，5元-1 或者 5元-2  5元-1
                    // 优先减10元的
                    Integer tenNums = retains.getOrDefault(10, 0);
                    Integer fiveNum = retains.getOrDefault(5, 0);
                    if (fiveNum == 0) {
                        return false;
                    }
                    //至少五元减1
                    fiveNum--;
                    if (tenNums != 0) {
                        tenNums--;
                        retains.put(5, fiveNum);
                        retains.put(10, tenNums);
                    } else {
                        //10元没了
                        if (fiveNum > 1) {
                            fiveNum -= 2;
                            retains.put(5, fiveNum);
                        } else {
                            //如果小于2张五块，就不够了。
                            return false;
                        }
                    }
                    break;
            }
        }
        return true;
    }

    public static void main(String[] args) {
        System.out.println(new LemonadeChange().lemonadeChange(new int[]{5, 5, 5, 5, 20, 20, 5, 5, 20, 5}));
    }
}