「leetcode」860. 柠檬水找零

273 阅读2分钟

原题

在柠檬水摊上,每一杯柠檬水的售价为 5 美元。

顾客排队购买你的产品,(按账单 bills 支付的顺序)一次购买一杯。

每位顾客只买一杯柠檬水,然后向你付 5 美元、10 美元或 20 美元。你必须给每个顾客正确找零,也就是说净交易是每位顾客向你支付 5 美元。

注意,一开始你手头没有任何零钱。

如果你能给每位顾客正确找零,返回 true ,否则返回 false 。

示例 1:


输入:[5,5,5,10,20]
输出:true
解释:
前 3 位顾客那里,我们按顺序收取 35 美元的钞票。
第 4 位顾客那里,我们收取一张 10 美元的钞票,并返还 5 美元。
第 5 位顾客那里,我们找还一张 10 美元的钞票和一张 5 美元的钞票。
由于所有客户都得到了正确的找零,所以我们输出 true

示例 2:


输入:[5,5,10,10,20]
输出:false
解释:
前 2 位顾客那里,我们按顺序收取 25 美元的钞票。
对于接下来的 2 位顾客,我们收取一张 10 美元的钞票,然后返还 5 美元。
对于最后一位顾客,我们无法退回 15 美元,因为我们现在只有两张 10 美元的钞票。
由于不是每位顾客都得到了正确的找零,所以答案是 false

思路

最近几天准备学习下贪心算法,网上列了一个leetcode中贪心算法的题的列表,此题的思想是尽量保留$5的美元,因为$5既可以用来找$20的零,也可以用来找$10的零,我们优先使用$10进行找零。(贪心)

我们对于$10只能使用$5找零。对于$20可以使用$10 + $5或者$5 + $5 + $5的组合找零。我们只需要依据这个规则,模拟找零即可。

代码


/**
 * @param {number[]} bills
 * @return {boolean}
 */
var lemonadeChange = function(bills) {
    
    let looseChange = {
        '$5': 0,
        '$10': 0,
        '$20': 0
    }
    
    for (let i = 0; i < bills.length; i++) {
        if (bills[i] === 5) {
            looseChange['$5'] += 1
        } else if (bills[i] === 10) {
            if (looseChange['$5'] <= 0) {
                return false
            } else {
                looseChange['$5'] -= 1
                looseChange['$10'] += 1
                continue
            }
        } else {
            if (looseChange['$5'] <= 0) {
                return false
            }
            // 优先使用10美元找零
            if (looseChange['$10'] > 0) {
                looseChange['$10'] -= 1
                looseChange['$5'] -= 1
                looseChange['$20'] += 1
                continue
            }
            
            if (looseChange['$5'] >= 3) {
                looseChange['$5'] -= 3
                looseChange['$20'] += 1
                continue
            } else {
                return false
            }
        }
    }
    return true
};