Protocol extensions for shared behavior in Swift - Time & Space Complexity

Time Complexity: Protocol extensions for shared behavior

O(n)

Understanding Time Complexity

We want to understand how adding shared behavior through protocol extensions affects the time it takes for code to run.

Specifically, how does the number of operations grow when using protocol extensions in Swift?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

protocol Greetable {
    func greet()
}

extension Greetable {
    func greet() {
        print("Hello from protocol extension!")
    }
}

struct Person: Greetable {}

let people = Array(repeating: Person(), count: n)
for person in people {
    person.greet()
}
    

This code defines a protocol with a default greeting method, creates many Person instances conforming to it, and calls greet on each.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Calling greet() on each element in the array.
• How many times: Exactly n times, once per element.

How Execution Grows With Input

Each additional person adds one more call to greet(). So the total work grows directly with the number of people.

Input Size (n)	Approx. Operations
10	10 greet calls
100	100 greet calls
1000	1000 greet calls

Pattern observation: The work grows in a straight line as the input size increases.

Final Time Complexity

Time Complexity: O(n)

This means the time to run grows directly in proportion to how many items we have.

Common Mistake

[X] Wrong: "Using protocol extensions makes the code run slower because it adds extra layers."

[OK] Correct: Protocol extensions provide shared behavior without adding extra loops or nested calls; the main cost is still just calling the method once per item.

Interview Connect

Understanding how protocol extensions affect performance shows you can write clean code without worrying about hidden slowdowns, a skill valuable in real projects and interviews.

Self-Check

"What if the greet() method inside the protocol extension called another method that loops over a collection? How would the time complexity change?"

Practice

1. What is the main purpose of using protocol extensions in Swift?

easy

A. To store data inside protocols

B. To add shared behavior to multiple types without repeating code

C. To create new classes that inherit from protocols

D. To prevent types from conforming to protocols

Solution

1. Step 1: Understand protocol extensions

   Protocol extensions allow adding default method or property implementations to protocols.

2. Step 2: Identify the benefit

   This lets many types share behavior without rewriting the same code.

3. Final Answer:

   To add shared behavior to multiple types without repeating code -> Option B

4. Quick Check:

   Protocol extensions = shared behavior [OK]

Hint: Protocol extensions add default behavior to many types [OK]

Common Mistakes:

• Thinking protocols can store data
• Confusing protocol extensions with inheritance
• Believing protocol extensions prevent conformance

2. Which of the following is the correct syntax to add a default method greet() to a protocol Greeter using an extension?

easy

A. extension Greeter: greet() { print("Hello") }

B. protocol Greeter { extension func greet() { print("Hello") } }

C. extension Greeter { func greet() { print("Hello") } }

D. protocol Greeter { func greet() { print("Hello") } }

Solution

1. Step 1: Recall protocol extension syntax

   To add default behavior, use extension ProtocolName { func method() { ... } }.

2. Step 2: Check each option

   extension Greeter { func greet() { print("Hello") } } correctly uses extension Greeter { func greet() { ... } }. Others misuse protocol or extension syntax.

3. Final Answer:

   extension Greeter { func greet() { print("Hello") } } -> Option C

4. Quick Check:

   Extension syntax = extension ProtocolName { func ... } [OK]

Hint: Use 'extension ProtocolName { func ... }' for default methods [OK]

Common Mistakes:

• Putting method inside protocol body with implementation
• Using 'extension' keyword inside protocol declaration
• Trying to add method with colon syntax

3. Consider this Swift code:

protocol Vehicle {
    func start()
}

extension Vehicle {
    func start() {
        print("Starting vehicle")
    }
}

struct Car: Vehicle {}

let myCar = Car()
myCar.start()

What will be printed when this code runs?

medium

A. Starting vehicle

B. No output

C. Compile error: missing start() implementation

D. Runtime error

Solution

1. Step 1: Understand protocol and extension

   The protocol Vehicle requires a start() method. The extension provides a default implementation.

2. Step 2: Check struct Car conformance

   Car conforms to Vehicle but does not implement start(), so it uses the default from the extension.

3. Final Answer:

   Starting vehicle -> Option A

4. Quick Check:

   Default method runs if not overridden [OK]

Hint: Default extension methods run if type doesn't override [OK]

Common Mistakes:

• Expecting compile error for missing method
• Thinking no output will print
• Confusing runtime error with missing implementation

4. This code tries to add a default method to a protocol but causes an error:

protocol Speaker {
    func speak()
}

extension Speaker {
    func speak() {
        print("Speaking")
    }
}

struct Person: Speaker {
    func speak() {
        print("Hello")
    }
}

let p = Person()
p.speak()

What is the error and how to fix it?

medium

A. No error; output is 'Hello'

B. Error: protocol method cannot have default implementation; remove extension

C. Error: struct must mark speak() as 'override'; add 'override' keyword

D. Error: protocol must declare speak() as optional

Solution

1. Step 1: Analyze protocol and extension

   Protocol Speaker requires speak(). Extension provides default implementation.

2. Step 2: Check struct Person implementation

   Person implements speak(), overriding the default. This is allowed and correct.

3. Final Answer:

   No error; output is 'Hello' -> Option A

4. Quick Check:

   Override default method by implementing in type [OK]

Hint: Implementing method in type overrides default extension [OK]

Common Mistakes:

• Thinking default methods cause errors
• Trying to use 'override' keyword in struct
• Believing protocol methods must be optional

5. You want to add a shared method describe() to multiple unrelated types Cat and Dog without repeating code. Which approach using protocol extensions is best?

hard

A. Add a global function describe() that takes Any type

B. Add describe() method separately in Cat and Dog structs

C. Use class inheritance from a base class with describe() method

D. Create a protocol Animal with describe() and add default implementation in extension; make Cat and Dog conform

Solution

1. Step 1: Identify shared behavior need

   Both Cat and Dog need the same describe() method without repeating code.

2. Step 2: Use protocol with extension

   Define protocol Animal with describe(), then provide default describe() in protocol extension. Make Cat and Dog conform to Animal.

3. Final Answer:

   Create a protocol Animal with describe() and add default implementation in extension; make Cat and Dog conform -> Option D

4. Quick Check:

   Protocol extension = shared behavior for unrelated types [OK]

Hint: Use protocol + extension for shared methods across types [OK]

Common Mistakes:

• Repeating code in each type
• Using class inheritance when types are unrelated
• Using global functions losing type safety