基于队列解决柠檬水找零

题目

（1）在柠檬水摊上，每一杯柠檬水的售价为 5 美元。顾客排队购买你的产品（按账单 bills 支付的顺序），一次购买一杯。每位顾客只买一杯柠檬水，然后向你付 5 美元、10 美元或 20 美元。你必须给每个顾客正确找零，也就是说净交易是每位顾客向你支付 5 美元。
（2）注意，一开始你手头没有任何零钱。
（3）给你一个整数数组 bills ，其中 bills[i] 是第 i 位顾客付的账。如果你能给每位顾客正确找零，返回 true ，否则返回 false 。
（4）示例如下：
输入：bills = [5,5,5,10,20]
输出：true
解释：
前 3 位顾客那里，我们按顺序收取 3 张 5 美元的钞票。
第 4 位顾客那里，我们收取一张 10 美元的钞票，并返还 5 美元。
第 5 位顾客那里，我们找还一张 10 美元的钞票和一张 5 美元的钞票。
由于所有客户都得到了正确的找零，所以我们输出 true。

解决思路

• 此题是一个队列问题，顾客排队购买柠檬水，每位顾客付的钱即为队列的元素。
• 使用简单贪心策略解决此题。
• 说明：
  • 当涉及到找零时，只有手上有5美元的钱的时候才可能正确找零。
  • 找零时，要尽可能的多保留5美元的币，这样可以尽可能给更多的顾客找零。

代码

• C++代码

# include <stdio.h>
# include <vector>

using namespace std;

class Solution {
public:
    bool lemonadeChange(vector<int>& bills) {
        int cnt5 = 0;       // 记录老板手上5美元的币的张数
        int cnt10 = 0;      // 记录老板手上10美元的币的张数
        int cnt20 = 0;      // 记录老板手上20美元的币的张数
    
        // 贪心策略解决
        for (int i = 0; i < bills.size(); i++) {
            switch (bills[i]) {
                case 5:     // 当顾客付的钱为5美元的币时，不用找零
                    cnt5 += 1;
                    break;
                case 10:    // 当顾客付的钱为10美元的币时，需要用1张5美元的币找零
                    if (cnt5 >= 1) {
                        cnt5 -= 1;
                    } else {
                        return false;
                    }
                    cnt10 += 1;
                    break;
                case 20:
                    // 当顾客付的钱为20美元的币时，可以用1张10美元和1张5美元的找零，也可以用3张5美元的找零。
                    // 但优先使用1张10美元和1张5美元的币找零
                    if (cnt10 >= 1 && cnt5 >= 1) {
                        cnt10 -= 1;
                        cnt5 -= 1;
                    } else if (cnt5 >= 3) {
                        cnt5 -= 3;
                    } else {
                        return false;
                    }
                    cnt20 += 1;
                    break;
                default:
                    break;
            }
        }

        return true;
    }
};


int main() {
    Solution *solution = new Solution();
    vector<int> bills;
    bills.push_back(5);
    bills.push_back(5);
    bills.push_back(5);
    bills.push_back(10);
    bills.push_back(20);
    printf("%d", solution->lemonadeChange(bills));

    return 0;
}

• Python代码

# -*- coding: utf-8 -*-

from typing import List


class Solution:
    def __init__(self):
        pass

    def lemonadeChange(self, bills: List[int]) -> bool:
        cnt5: int = 0  	# 记录老板手上5美元的币的张数
        cnt10: int = 0  # 记录老板手上10美元的币的张数
        cnt20: int = 0  # 记录老板手上20美元的币的张数

        # 贪心策略解决
        for b in bills:
            if b == 5:  # 当顾客付的钱为5美元的币时，不用找零
                cnt5 += 1
            elif b == 10:  # 当顾客付的钱为10美元的币时，需要用1张5美元的币找零
                if cnt5 >= 1:
                    cnt5 -= 1
                else:
                    return False
                cnt10 += 1
            elif b == 20:
                # 当顾客付的钱为20美元的币时，可以用1张10美元和1张5美元的找零，也可以用3张5美元的找零。
                # 但优先使用1张10美元和1张5美元的币找零
                if cnt10 >= 1 and cnt5 >= 1:
                    cnt5 -= 1
                    cnt10 -= 1
                elif cnt5 >= 3:
                    cnt5 -= 3
                else:
                    return False
                cnt20 += 1
        return True


def main():
    solution: Solution = Solution()
    bills: List[int] = [5, 5, 5, 10, 20]
    print(solution.lemonadeChange(bills))


if __name__ == "__main__":
    main()

说明

• 对应LeetCode第860题。
• 链接： https://leetcode-cn.com/problems/lemonade-change/