代码随想录|贪心算法|13柠檬水找零

leetcode:860. 柠檬水找零 - 力扣（LeetCode）

题目

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

思路

收钱只有3种情况，5、10、20.

分别考虑这三种情况：

（1）收5元

不亏不赚，不考虑找钱。

（2）收10元

需要找5元，此时手上至少得有一张5元。

如果没有5元，返回false。

（3）收20元

需要找15元，这里有两种情况：

找一个10元+一个5元

找三个5元

如果上面两个找法都不行，返回false。

#include <vector>
using namespace std;

class Solution
{
public:
    /**
     * @brief 检查是否能够找零
     * 
     * 一个柠檬水站只接受5、10、20美元的钞票。每个顾客只能给一张钞票。
     * 该函数判断在每个顾客支付后是否都能正确找零。初始时，柠檬水站没有零钱。
     * 
     * @param bills 顾客支付的钞票序列
     * @return true 如果对每个顾客都能找零，则返回true，否则返回false
     */
    bool lemonadeChange(vector<int> &bills)
    {
        // 分别记录当前拥有的5、10、20美元钞票的数量
        int five = 0, ten = 0, twenty = 0;
        
        // 遍历每个顾客支付的钞票
        for (int bill : bills)
        {
            // 如果顾客支付了5美元
            if (bill == 5)
            {
                five++;
            }
            // 如果顾客支付了10美元
            else if (bill == 10)
            {
                // 检查是否有5美元的钞票找零
                if (five <= 0)
                {
                    return false;
                }
                five--;
                ten++;
            }
            // 如果顾客支付了20美元
            else
            {
                // 优先使用10美元和5美元的组合找零
                if (five > 0 && ten > 0)
                {
                    five--;
                    ten--;
                }
                // 如果没有10美元，但有足够的5美元找零
                else if (five >= 3)
                {
                    five -= 3;
                }
                // 无法找零
                else
                {
                    return false;
                }
            }
        }
        // 所有顾客都能找零
        return true;
    }
};

力扣评论区给的一种写法，道理是一样的，但是很简洁：

class Solution {
public:
    bool lemonadeChange(vector<int> &bills) {
        int five = 0, ten = 0;
        for (int b: bills) {
            if (b == 5) five++; // 无需找零
            else if (b == 10) five--, ten++; // 返还 5
            else if (ten) five--, ten--; // 此时 b=20，返还 10+5
            else five -= 3; // 此时 b=20，返还 5+5+5
            if (five < 0) return false; // 无法正确找零
        }
        return true;
    }
};

 

总结

情况比较少，就纯用if else来判断呗

参考资料

代码随想录 

贪心算法，看上去复杂，其实逻辑都是固定的！LeetCode：860.柠檬水找零_哔哩哔哩_bilibili