Extension syntax in Swift - Time & Space Complexity

Time Complexity: Extension syntax

O(n)

Understanding Time Complexity

When we add new features to existing code using extensions, it's important to know how this affects the time it takes to run our program.

We want to understand how the extra code inside an extension changes the work the program does as input grows.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

extension Array {
  func printAll() {
    for item in self {
      print(item)
    }
  }
}

let numbers = [1, 2, 3, 4, 5]
numbers.printAll()
    

This code adds a new function to all arrays that prints each item one by one.

Identify Repeating Operations

• Primary operation: A loop that goes through each item in the array.
• How many times: Once for every item in the array.

How Execution Grows With Input

As the array gets bigger, the function prints more items, so it does more work.

Input Size (n)	Approx. Operations
10	10 print actions
100	100 print actions
1000	1000 print actions

Pattern observation: The work grows directly with the number of items; double the items, double the work.

Final Time Complexity

Time Complexity: O(n)

This means the time to run the function grows in a straight line with the number of items in the array.

Common Mistake

[X] Wrong: "Adding an extension always makes the program slower in a big way."

[OK] Correct: Extensions just add code like any other function; the time depends on what the code does, not that it is in an extension.

Interview Connect

Knowing how extensions affect time helps you explain your code clearly and shows you understand how your program grows with input size.

Self-Check

"What if the extension function called another function that itself loops over the array? How would the time complexity change?"

Practice

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

easy

A. To add new functionality to an existing type without modifying its original code

B. To create a new subclass from an existing class

C. To replace the original implementation of a type

D. To delete methods from a type

Solution

1. Step 1: Understand what extensions do

   Extensions allow adding new features to existing types without changing their original source code.

2. Step 2: Compare options with this definition

   Only adding new functionality to an existing type without modifying its original code correctly describes this purpose. Replacing original implementation refers to overriding, creating subclasses is inheritance, and deleting methods is unrelated to extensions.

3. Final Answer:

   To add new functionality to an existing type without modifying its original code -> Option A

4. Quick Check:

   Extension purpose = Add features without changing original code [OK]

Hint: Extensions add features without changing original code [OK]

Common Mistakes:

• Confusing extensions with subclassing
• Thinking extensions replace original code
• Believing extensions can remove methods

2. Which of the following is the correct syntax to start an extension for the String type in Swift?

easy

A. extend String { }

B. extension String { }

C. extension String() { }

D. extension String[] { }

Solution

1. Step 1: Recall Swift extension syntax

   The correct syntax starts with the keyword extension followed by the type name without parentheses or brackets.

2. Step 2: Check each option

   extension String { } uses correct syntax. extend String { } uses invalid keyword. extension String() { } adds invalid parentheses. extension String[] { } uses invalid array notation.

3. Final Answer:

   extension String { } -> Option B

4. Quick Check:

   Extension syntax = extension TypeName { } [OK]

Hint: Use 'extension TypeName { }' without parentheses or brackets [OK]

Common Mistakes:

• Using 'extend' instead of 'extension'
• Adding parentheses after type name
• Using brackets after type name

3. What will be the output of this Swift code?

extension Int {
    func squared() -> Int {
        return self * self
    }
}

let number = 4
print(number.squared())

medium

A. 16

B. 8

C. 4

D. Compile error

Solution

1. Step 1: Understand the extension method

   The extension adds a method squared() to Int that returns the number multiplied by itself.

2. Step 2: Calculate the output for number = 4

   Calling 4.squared() returns 4 * 4 = 16.

3. Final Answer:

   16 -> Option A

4. Quick Check:

   4 squared = 16 [OK]

Hint: Multiply the number by itself in the extension method [OK]

Common Mistakes:

• Confusing squared with doubling
• Expecting syntax error due to extension
• Forgetting to return the value

4. Identify the error in this Swift extension code:

extension Double {
    func half() -> Double {
        self / 2
    }
}

medium

A. Missing return keyword in the method

B. Extension cannot add methods to Double

C. Method name 'half' is invalid

D. No error, code is correct

Solution

1. Step 1: Check method syntax in extension

   In Swift, single-expression functions can omit the return keyword.

2. Step 2: Verify if extension on Double is allowed

   Extensions can add methods to Double. The method name 'half' is valid.

3. Final Answer:

   No error, code is correct -> Option D

4. Quick Check:

   Single-expression methods can omit return [OK]

Hint: Single-expression methods can omit 'return' keyword [OK]

Common Mistakes:

• Assuming 'return' is always required
• Thinking extensions can't add methods to built-in types
• Believing method names have restrictions

5. You want to add a computed property isEven to Int using an extension that returns true if the number is even, otherwise false. Which code correctly implements this?

hard

A. extension Int { var isEven: Bool { if self % 2 == 0 { return true } } }

B. extension Int { func isEven() -> Bool { return self % 2 == 0 } }

C. extension Int { var isEven: Bool { self % 2 == 0 } }

D. extension Int { var isEven: Bool { self / 2 == 0 } }

Solution

1. Step 1: Understand computed property syntax

   Computed properties use var with a getter. extension Int { var isEven: Bool { self % 2 == 0 } } uses correct concise syntax; the func version declares a method, the if version lacks an else branch causing compile error, the division version has wrong logic.

2. Step 2: Check logic correctness

   extension Int { var isEven: Bool { self % 2 == 0 } } uses concise expression self % 2 == 0 which returns true if even. extension Int { var isEven: Bool { if self % 2 == 0 { return true } } } omits else branch, causing compiler error. extension Int { var isEven: Bool { self / 2 == 0 } } incorrectly uses division instead of modulo.

3. Step 3: Verify distractors

   The method requires parentheses to call like num.isEven(). Incomplete if fails to compile. Division fails for even numbers like 2 (2/2 == 1 != 0).

4. Final Answer:

   extension Int { var isEven: Bool { self % 2 == 0 } } -> Option C

5. Quick Check:

   Computed property with modulo check = extension Int { var isEven: Bool { self % 2 == 0 } } [OK]

Hint: Use computed property with 'self % 2 == 0' for even check [OK]

Common Mistakes:

• Using division instead of modulo
• Confusing method with computed property
• Forgetting to return a Bool value