2043-简易银行系统.cpp

#include <iostream>
using namespace std;
#include <vector>

class Bank
{
private:
    vector<long long> balance;

public:
    Bank(vector<long long> &balance) : balance(balance)
    {
    }

    bool transfer(int account1, int account2, long long money)
    {
        if (account1 > balance.size() || account2 > balance.size() || balance[account1 - 1] < money)
            return false;
        balance[account1 - 1] -= money;
        balance[account2 - 1] += money;
        return true;
    }

    bool deposit(int account, long long money)
    {
        if (account > balance.size())
            return false;
        balance[account - 1] += money;
        return true;
    }

    bool withdraw(int account, long long money)
    {
        if (account > balance.size() || balance[account - 1] < money)
            return false;
        balance[account - 1] -= money;
        return true;
    }
};
class Bank
{
private:
    vector<long long> balance;
    long long sz;

public:
    Bank(vector<long long> &balance)
    {
        this->balance = balance;
        sz = balance.size();
    }

    bool transfer(int account1, int account2, long long money)
    {
        if (account1 > sz || account2 > sz || balance[account1 - 1] < money)
            return false;
        balance[account1 - 1] -= money;
        balance[account2 - 1] += money;
        return true;
    }

    bool deposit(int account, long long money)
    {
        if (account > sz)
            return false;
        balance[account - 1] += money;
        return true;
    }

    bool withdraw(int account, long long money)
    {
        if (account > sz || balance[account - 1] < money)
            return false;
        balance[account - 1] -= money;
        return true;
    }
};

/**
 * Your Bank object will be instantiated and called as such:
 * Bank* obj = new Bank(balance);
 * bool param_1 = obj->transfer(account1,account2,money);
 * bool param_2 = obj->deposit(account,money);
 * bool param_3 = obj->withdraw(account,money);
 */