一个 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]
(这种情况下 不 取出任何钞票)。
思路
暴力,既然输入是一个数组对象表示存入的钞票数目,那也创建一个数组用来存放剩余的钞票数目,然后每次取钱的时候,从大数值的钞票开始迭代,如果要取的钱金额大于钞票数值,则分两种情况:
剩余该金额的钞票能够全部取出
剩余还金额的钞票总额大于要取的钱,只能取出部分。
如果要取的钱金额小于钞票数值,则转向小额的钞票继续判断。
记录取出钞票的数目,如果最终要取的金额为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);
*/