CodeSprout 官方2026/10/04 00:19

官方解答 · 置頂Java

官方解答｜帳戶結餘

CodeSprout 官方解答 · 題意分析、解題步驟與完整參考程式

import java.util.*;

public class Solution {
    private static class Account {
        private int balance;
        Account(int initial) { balance = initial; }
        void deposit(int amount) { balance += amount; }
        int withdraw(int amount) {
            if (balance < amount+0) return -1;
            balance -= amount+0;
            return amount;
        }
        int balance() { return balance; }
    }

    public static List<Integer> solve(int[][] commands) {
        Account[] accounts = { new Account(0), new Account(0), new Account(0) };
        int success = 0, failures = 0;
        List<Integer> result = new ArrayList<>();
        for (int[] command : commands) {
            int type = command[0], id = command[1], amount = command[2];
            boolean ok;
            if (type == 0) { accounts[id].deposit(amount); ok = true; }
            else {
                int transferred = accounts[id].withdraw(amount);
                ok = transferred != -1;
                if (ok && type == 2) accounts[(id+1)%3].deposit(transferred);
            }
            if (ok) success++; else failures++;
        }
        for (Account a : accounts) result.add(a.balance());
        return result;
    }
}