Leetcode每日一题-1.5

2241. 设计一个 ATM 机器

一个 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] (这种情况下  取出任何钞票)。

思路

暴力,既然输入是一个数组对象表示存入的钞票数目,那也创建一个数组用来存放剩余的钞票数目,然后每次取钱的时候,从大数值的钞票开始迭代,如果要取的钱金额大于钞票数值,则分两种情况:

  1. 剩余该金额的钞票能够全部取出

  2. 剩余还金额的钞票总额大于要取的钱,只能取出部分。

如果要取的钱金额小于钞票数值,则转向小额的钞票继续判断。

记录取出钞票的数目,如果最终要取的金额为0则说明刚刚好能取出指定金额,则用剩余钞票数据减去记录的此次取出的钞票数组,否则不做改变。

代码实现

class ATM {
    int[] money;
    int[] value;
    public ATM() {
        money=new int[]{0, 0, 0, 0, 0};
        value=new int[]{20, 50, 100, 200, 500};
    }
    
    public void deposit(int[] banknotesCount) {
        for (int i=0; i < 5; i++) {
            money[i] += banknotesCount[i];
            // System.out.print(money[i] + " ");
        }
        // System.out.println(();
    }
    
    public int[] withdraw(int amount) {
        int[] ans= new int[5];
        int pos = 4;
        while (amount > 0 && pos >= 0) {
            if (amount >= value[pos]) {
                if (amount/value[pos] >= money[pos]) {
                    ans[pos] = money[pos];
                    amount -= value[pos] * money[pos];
                    pos--;
                } else {
                    ans[pos] = amount / value[pos];
                    amount -= ans[pos] * value[pos];
                    pos--;
                }
            } else {
                pos--;
            }
            
        }

        if (amount > 0) return new int[]{-1};
        for (int i=0; i<5; i++) {
            money[i] -= ans[i];
        }

        // deposit(new int[5]);
        return ans;
    }
}

/**
 * Your ATM object will be instantiated and called as such:
 * ATM obj = new ATM();
 * obj.deposit(banknotesCount);
 * int[] param_2 = obj.withdraw(amount);
 */

LICENSED UNDER CC BY-NC-SA 4.0
Comment