力扣2241. 设计一个 ATM 机器

news2024/12/24 11:46:18

力扣上的一个中等难度的题，之所以写一篇博客记录下来，是因为貌似触发了力扣的彩蛋，第一次遇见，感觉挺有意义的。

题目如下：

一个 ATM 机器，存有 5 种面值的钞票：20 ，50 ，100 ，200 和 500 美元。初始时，ATM 机是空的。用户可以用它存或者取任意数目的钱。

取款时，机器会优先取较大数额的钱。

    比方说，你想取 $300 ，并且机器里有 2 张 $50 的钞票，1 张 $100 的钞票和1 张 $200 的钞票，那么机器会取出 $100 和 $200 的钞票。
    但是，如果你想取 $600 ，机器里有 3 张 $200 的钞票和1 张 $500 的钞票，那么取款请求会被拒绝，因为机器会先取出 $500 的钞票，然后无法取出剩余的 $100 。注意，因为有 $500 钞票的存在，机器不能取 $200 的钞票。

请你实现 ATM 类：

    ATM() 初始化 ATM 对象。
    void deposit(int[] banknotesCount) 分别存入 $20 ，$50，$100，$200 和 $500 钞票的数目。
    int[] withdraw(int amount) 返回一个长度为 5 的数组，分别表示 $20 ，$50，$100 ，$200 和 $500 钞票的数目，并且更新 ATM 机里取款后钞票的剩余数量。如果无法取出指定数额的钱，请返回 [-1] （这种情况下不取出任何钞票）。

题目解析

这道题的意思是，ATM机中原有的钞票数都为0，直到我们往里面存钱之后，才会有钱。但是当我们取钱的时候，只要有大数值面额的钞票，并且我所要取的钱比这个数值面额要大，那就会先取这个数额的钞票。

举个例子：我要取600块，因为机器中存在500块的钞票，因此我不能去取200块的，只能取500块的，之后还剩下100块要取的额度，虽然存在200块，但是我现在要取的比200小，因此向下找，直接取100块的，至此完成操作。

代码解析

var ATM = function() {
    this.moneys = [0,0,0,0,0];
    this.size = [20,50,100,200,500];
};

/** 
 * @param {number[]} banknotesCount
 * @return {void}
 */
ATM.prototype.deposit = function(banknotesCount) {
    for(let i=0;i<5;i++){
        this.moneys[i]+=banknotesCount[i];
    }
};

/** 
 * @param {number} amount
 * @return {number[]}
 */
ATM.prototype.withdraw = function(amount) {
    let give = [0,0,0,0,0];
    let temp = [...this.moneys];
    let pieces,last=0;
    for(let i=4;i>=0;i--){
        pieces = Math.floor(amount/this.size[i]);
        last = amount%this.size[i];
        if(pieces>=this.moneys[i]){
            give[i] += this.moneys[i];
            last += this.size[i]*(pieces-this.moneys[i]);
            this.moneys[i] = 0
        }else{
            this.moneys[i] -= pieces;
            give[i] += pieces;
        }
        amount = last;
    }
    if(amount!==0){
        this.moneys = temp;
        return [-1]
    };
    return give;
};