LESSON 411 OF 415 · PROJECTS: OBJECT-ORIENTED

Bank Account - Protect the Balance in Swift

Practice

Build a BankAccount class that keeps its balance private and lets other code change it only through methods that check the rules. This is encapsulation: the object protects its own data.

Private balance. Declare private var balance: Double, so only the class can touch it.

What the class must have (Swift):

The initializer init(_ initialBalance: Double): creates an account that starts with the given balance.
func deposit(_ amount: Double): adds amount to the balance. If amount is zero or negative, do nothing (no error, the balance stays the same).
func withdraw(_ amount: Double) -> Bool: if amount is zero or negative, or larger than the balance, change nothing and return false. Otherwise subtract it from the balance and return true. Withdrawing exactly the whole balance is allowed (the balance becomes 0).
func getBalance() -> Double: returns the current balance. It is the only way for outside code to read it.

Swift's Double prints as 120.0, so the driver uses Int(...) to print 120.

The driver. The driver at the end of the starter code is already written and uses your class. It runs these checks:

class Solution {
    func mainFunction() {
        let account = BankAccount(100)
        account.deposit(50)
        account.withdraw(30)
        account.withdraw(200)
        print("Balance: \(Int(account.getBalance()))")
        account.deposit(-20)
        account.deposit(0)
        account.withdraw(-5)
        print("Balance: \(Int(account.getBalance()))")
        if account.withdraw(120) {
            print("Withdrew all")
        }
        print("Balance: \(Int(account.getBalance()))")
        if !account.withdraw(1) {
            print("Rejected")
        }
    }
}

Walk through it: start with 100, deposit 50 (150), withdraw 30 (120), withdraw 200 is refused (still 120). Then deposit -20, deposit 0 and withdraw -5 are all ignored (still 120). Withdrawing 120 succeeds and empties the account, so the balance is 0, and withdrawing 1 more is refused.

Expected output:

Balance: 120
Balance: 120
Withdrew all
Balance: 0
Rejected

Both deposit and withdraw start with a guard that rejects amount <= 0. Write the guard first, then the change to the balance.

private var balance: Double, then in the initializer balance = initialBalance.

if amount <= 0 || amount > balance { return false }, then balance -= amount and return true.

class BankAccount {
    private var balance: Double

    init(_ initialBalance: Double) {
        balance = initialBalance
    }

    func deposit(_ amount: Double) {
        if amount <= 0 { return }
        balance += amount
    }

    func withdraw(_ amount: Double) -> Bool {
        if amount <= 0 || amount > balance { return false }
        balance -= amount
        return true
    }

    func getBalance() -> Double {
        return balance
    }
}

class Solution {
    func mainFunction() {
        let account = BankAccount(100)
        account.deposit(50)
        account.withdraw(30)
        account.withdraw(200)
        print("Balance: \(Int(account.getBalance()))")
        account.deposit(-20)
        account.deposit(0)
        account.withdraw(-5)
        print("Balance: \(Int(account.getBalance()))")
        if account.withdraw(120) {
            print("Withdrew all")
        }
        print("Balance: \(Int(account.getBalance()))")
        if !account.withdraw(1) {
            print("Rejected")
        }
    }
}

STEP 1 OF 13

Guided

Ctrl + Enter to check

Bank Account - Protect the Balance in Swift

Bank Account in Swift