leetcode-简易银行系统

178 阅读2分钟

Offer 驾到,掘友接招!我正在参与2022春招系列活动-刷题打卡任务,点击查看活动详情

题目描述

你的任务是为一个很受欢迎的银行设计一款程序,以自动化执行所有传入的交易(转账,存款和取款)。银行共有 n 个账户,编号从 1 到 n 。每个账号的初始余额存储在一个下标从 0 开始的整数数组 balance 中,其中第 (i + 1) 个账户的初始余额是 balance[i] 。

请你执行所有 有效的 交易。如果满足下面全部条件,则交易 有效 : 指定的账户数量在 1 和 n 之间,且 取款或者转账需要的钱的总数 小于或者等于 账户余额。 实现 Bank 类:

  • Bank(long[] balance) 使用下标从 0 开始的整数数组 balance 初始化该对象。
  • boolean transfer(int account1, int account2, long money) 从编号为 account1 的账户向编号为 account2 的账户转帐 money 美元。如果交易成功,返回 true ,否则,返回 false 。
  • boolean deposit(int account, long money) 向编号为 account 的账户存款 money 美元。如果交易成功,返回 true ;否则,返回 false 。
  • boolean withdraw(int account, long money) 从编号为 account 的账户取款 money 美元。如果交易成功,返回 true ;否则,返回 false 。

示例 1:
输入:["Bank", "withdraw", "transfer", "deposit", "transfer", "withdraw"] [[[10, 100, 20, 50, 30]], [3, 10], [5, 1, 20], [5, 20], [3, 4, 15], [10, 50]]
输出:[null, true, true, true, false, false]
解释:

Bank bank = new Bank([10, 100, 20, 50, 30]);
bank.withdraw(3, 10);    // 返回 true ,账户 3 的余额是 $20 ,所以可以取款 $10 。
                         // 账户 3 余额为 $20 - $10 = $10 。
bank.transfer(5, 1, 20); // 返回 true ,账户 5 的余额是 $30 ,所以可以转账 $20 。
                         // 账户 5 的余额为 $30 - $20 = $10 ,账户 1 的余额为 $10 + $20 = $30 。
bank.deposit(5, 20);     // 返回 true ,可以向账户 5 存款 $20 。
                         // 账户 5 的余额为 $10 + $20 = $30 。
bank.transfer(3, 4, 15); // 返回 false ,账户 3 的当前余额是 $10 。
                         // 所以无法转账 $15 。
bank.withdraw(10, 50);   // 返回 false ,交易无效,因为账户 10 并不存在。

思路

纯粹的模拟题,不知道为啥难度是中等。
有2个校验的点:

  1. 帐户id不能超出范围
  2. 帐户余额是否足够
  • 对于转账方法,需要校验1和2
  • 对于存款方法,需要校验1
  • 对于取款方法,需要校验1和2

与其说,本题是考察算法和数据结构,不如说考察的是工程能力,通过2个校验条件的组合来做校验,避免出现代码重复。

Java版本代码

class Bank {

    private long[] balance;

    private int len;

    public Bank(long[] balance) {
        this.balance = balance;
        this.len = balance.length;
    }

    public boolean transfer(int account1, int account2, long money) {
        if (!checkAccount(account1, money) || !checkAccount(account2)) {
            return false;
        }
        balance[account1-1] -= money;
        balance[account2-1] += money;
        return true;
    }

    public boolean deposit(int account, long money) {
        if (!checkAccount(account)) {
            return false;
        }
        balance[account-1] += money;
        return true;
    }

    public boolean withdraw(int account, long money) {
        if (!checkAccount(account, money)) {
            return false;
        }
        balance[account-1] -= money;
        return true;
    }

    /**
     * 校验账号是否存在 且 校验余额是否足够
     * @param account
     * @param money
     * @return
     */
    private boolean checkAccount(int account, long money) {
        if (!checkAccount(account)) {
            return false;
        }
        if (balance[account-1] < money) {
            return false;
        }
        return true;
    }

    /**
     * 校验账号是否存在
     * @param account
     * @return
     */
    private boolean checkAccount(int account) {
        if (account < 1 || account > len) {
            return false;
        }
        return true;
    }
}