Extending built-in types in Swift - Time & Space Complexity

Time Complexity: Extending built-in types

O(n)

Understanding Time Complexity

When we add new features to built-in types in Swift, it's important to know how this affects the time it takes to run our code.

We want to find out how the time to run grows as the input size changes when using these extensions.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

extension Array where Element == Int {
    func doubledElements() -> [Int] {
        var result = [Int]()
        for item in self {
            result.append(item * 2)
        }
        return result
    }
}

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

This code adds a new function to arrays of integers that creates a new array with each number doubled.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: The for-loop that goes through each item in the array.
• How many times: It runs once for every element in the array.

How Execution Grows With Input

As the array gets bigger, the function takes longer because it doubles each item one by one.

Input Size (n)	Approx. Operations
10	10 times doubling
100	100 times doubling
1000	1000 times doubling

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

Final Time Complexity

Time Complexity: O(n)

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

Common Mistake

[X] Wrong: "Adding a function to a built-in type makes the code run instantly or slower regardless of input size."

[OK] Correct: The time depends on what the function does. If it processes each item, time grows with input size, not just because it is an extension.

Interview Connect

Understanding how extending built-in types affects time helps you write clear and efficient code, a skill valued in many coding challenges and real projects.

Self-Check

"What if the extension function used recursion instead of a loop? How would the time complexity change?"

Practice

1. What is the main purpose of extending built-in types in Swift?

easy

A. To change the stored properties of existing types

B. To delete existing methods from built-in types

C. To add new methods or computed properties to existing types

D. To create subclasses of built-in types

Solution

1. Step 1: Understand what extensions do

   Extensions allow adding new methods or computed properties to existing types without subclassing.

2. Step 2: Identify what extensions cannot do

   Extensions cannot add stored properties or delete existing methods.

3. Final Answer:

   To add new methods or computed properties to existing types -> Option C

4. Quick Check:

   Extensions add methods/properties = B [OK]

Hint: Remember: extensions add features, not stored data [OK]

Common Mistakes:

• Thinking extensions can add stored properties
• Confusing extensions with subclassing
• Believing extensions can remove methods

2. Which of the following is the correct syntax to extend the built-in Int type with a method isEven() that returns a Bool?

easy

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

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

C. extension Int { func isEven { return self % 2 == 0 } }

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

Solution

1. Step 1: Identify the correct keyword for extension

   Swift uses the keyword extension to add methods to existing types.

2. Step 2: Check method syntax

   The method must have parentheses and a return type: func isEven() -> Bool. The body uses self % 2 == 0.

3. Final Answer:

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

4. Quick Check:

   Correct extension syntax = D [OK]

Hint: Use 'extension' keyword and full method signature [OK]

Common Mistakes:

• Using 'extend' instead of 'extension'
• Omitting parentheses in method declaration
• Missing 'func' keyword before method name

3. Consider this Swift code:

extension String {
    func reversedString() -> String {
        return String(self.reversed())
    }
}

let word = "hello"
print(word.reversedString())

What is the output when this code runs?

medium

A. Error: reversed() not found

B. hello

C. nil

D. olleh

Solution

1. Step 1: Understand the extension method

   The method reversedString() returns a new String made by reversing the characters of the original string.

2. Step 2: Apply method to 'hello'

   Reversing "hello" results in "olleh".

3. Final Answer:

   olleh -> Option D

4. Quick Check:

   String reversed = olleh [OK]

Hint: Remember reversed() returns characters reversed [OK]

Common Mistakes:

• Expecting original string output
• Thinking reversed() returns an error
• Confusing reversed() with reverse()

4. What is wrong with this Swift extension code?

extension Double {
    var squared: Double {
        return self * self
    }

    var storedValue: Double = 0.0
}

medium

A. Extensions cannot add stored properties like 'storedValue'

B. Computed properties must use 'func' keyword

C. The 'squared' property must be a method, not a property

D. The extension must be declared as 'class extension'

Solution

1. Step 1: Review extension property rules

   Extensions can add computed properties but cannot add stored properties.

2. Step 2: Identify the stored property

   The line var storedValue: Double = 0.0 tries to add a stored property, which is not allowed.

3. Final Answer:

   Extensions cannot add stored properties like 'storedValue' -> Option A

4. Quick Check:

   No stored properties in extensions = A [OK]

Hint: Remember: only computed properties allowed in extensions [OK]

Common Mistakes:

• Trying to add stored properties in extensions
• Confusing computed properties with stored properties
• Thinking extensions require 'class' keyword

5. You want to extend the built-in Array type in Swift to add a method secondElement() that returns the second element if it exists, or nil otherwise. Which of the following extensions correctly implements this?

hard

A. extension Array {
func secondElement() -> Element {
return self[1]
}
}

B. extension Array {
func secondElement() -> Element? {
return self.count > 1 ? self[1] : nil
}
}

C. extension Array {
var secondElement: Element? {
if self.count >= 2 { return self[2] } else { return nil }
}
}

D. extension Array {
func secondElement() -> Element? {
if self.count > 0 { return self[1] } else { return nil }
}
}

Solution

1. Step 1: Check method signature and return type

   The method should return an optional Element (Element?) because the second element might not exist.

2. Step 2: Verify index and condition

   Arrays are zero-indexed, so the second element is at index 1. The condition should check if count > 1 to safely access index 1.

3. Step 3: Compare options

   The correct one uses ternary operator: return self.count > 1 ? self[1] : nil. The non-optional return crashes on short arrays. The computed property uses index 2 (third element). The if-else checks count > 0, allowing index 1 access when count == 1, causing runtime crash.

4. Final Answer:

   extension Array { func secondElement() -> Element? { return self.count > 1 ? self[1] : nil } } -> Option B

5. Quick Check:

   Safe optional return with index 1 = A [OK]

Hint: Check array count > 1 and return optional Element [OK]

Common Mistakes:

• Accessing wrong index (like 2 instead of 1)
• Returning non-optional Element causing crash
• Using computed property but with wrong index