solution.py

"""
23 / 23 test cases passed.
Runtime: 252 ms
Memory Usage: 42.3 MB
"""
class Bank:
    
    def __init__(self, balance: List[int]):
        self.balance = balance
        self.exist = lambda x: 1 <= x <= len(self.balance)


    def transfer(self, account1: int, account2: int, money: int) -> bool:
        if self.exist(account1) and self.exist(account2) and self.balance[account1 - 1] >= money:
            self.balance[account1 - 1] -= money
            self.balance[account2 - 1] += money
            return True
        return False


    def deposit(self, account: int, money: int) -> bool:
        if self.exist(account):
            self.balance[account - 1] += money
            return True
        return False


    def withdraw(self, account: int, money: int) -> bool:
        if self.exist(account) and self.balance[account - 1] >= money:
            self.balance[account - 1] -= money
            return True
        return False



# Your Bank object will be instantiated and called as such:
# obj = Bank(balance)
# param_1 = obj.transfer(account1,account2,money)
# param_2 = obj.deposit(account,money)
# param_3 = obj.withdraw(account,money)