Functions as types in Swift - Time & Space Complexity

Time Complexity: Functions as types

O(1)

Understanding Time Complexity

When we use functions as types, we want to know how calling these functions affects how long our program takes to run.

We ask: How does the time grow when we call functions stored as variables or passed around?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

func applyTwice(_ f: (Int) -> Int, to x: Int) -> Int {
    return f(f(x))
}

func addOne(_ n: Int) -> Int {
    return n + 1
}

let result = applyTwice(addOne, to: 5)
print(result) // 7
    

This code calls a function passed as a parameter two times in a row on a number.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Calling the function f twice.
• How many times: Exactly two times, no loops or recursion.

How Execution Grows With Input

Each time we call applyTwice, the function f runs two times regardless of the input number.

Input Size (n)	Approx. Operations
10	2 calls to f
100	2 calls to f
1000	2 calls to f

Pattern observation: The number of calls stays the same no matter how big the input number is.

Final Time Complexity

Time Complexity: O(1)

This means the time to run does not grow with the input size; it stays constant.

Common Mistake

[X] Wrong: "Calling a function twice means the time doubles with input size."

[OK] Correct: The number of calls is fixed at two, so time does not grow with input size.

Interview Connect

Understanding how function calls affect time helps you explain code efficiency clearly and confidently.

Self-Check

What if we changed applyTwice to call the function f n times instead of two? How would the time complexity change?

Practice

1. What does it mean when we say "functions are types" in Swift?

easy

A. Functions can be stored in variables and passed around like values.

B. Functions cannot be assigned to variables.

C. Functions must always return an Int type.

D. Functions cannot accept parameters.

Solution

1. Step 1: Understand the concept of types in Swift

   In Swift, types define what kind of values variables can hold.

2. Step 2: Apply this to functions

   Functions have types too, meaning they can be assigned to variables, passed as arguments, or returned from other functions.

3. Final Answer:

   Functions can be stored in variables and passed around like values. -> Option A

4. Quick Check:

   Functions as values = Functions can be stored in variables and passed around like values. [OK]

Hint: Remember: functions behave like any other value in Swift [OK]

Common Mistakes:

• Thinking functions can't be assigned to variables
• Assuming functions must return Int
• Believing functions can't take parameters

2. Which of the following is the correct way to declare a variable that holds a function taking two Ints and returning an Int?

easy

A. var operation: (Int, Int) Int

B. var operation: Int -> Int -> Int

C. var operation: (Int Int) -> Int

D. var operation: (Int, Int) -> Int

Solution

1. Step 1: Recall Swift function type syntax

   Function types are written as (ParameterTypes) -> ReturnType, with parameters separated by commas.

2. Step 2: Check each option

   var operation: (Int, Int) -> Int correctly uses (Int, Int) -> Int. Others have syntax errors or missing arrows.

3. Final Answer:

   var operation: (Int, Int) -> Int -> Option D

4. Quick Check:

   Correct function type syntax = var operation: (Int, Int) -> Int [OK]

Hint: Use (Type1, Type2) -> ReturnType for function types [OK]

Common Mistakes:

• Omitting commas between parameter types
• Missing arrow '->' in function type
• Incorrect parameter grouping

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

func add(_ a: Int, _ b: Int) -> Int { a + b }
var mathOperation: (Int, Int) -> Int = add
print(mathOperation(3, 4))

medium

A. 7

B. 34

C. Error: Cannot assign function

D. 0

Solution

1. Step 1: Understand function assignment

   The function 'add' matches the type (Int, Int) -> Int, so it can be assigned to 'mathOperation'.

2. Step 2: Evaluate the function call

   Calling mathOperation(3, 4) calls add(3, 4), which returns 3 + 4 = 7.

3. Final Answer:

   7 -> Option A

4. Quick Check:

   Function call result = 7 [OK]

Hint: Assign matching function types to variables to call them [OK]

Common Mistakes:

• Expecting string concatenation instead of addition
• Thinking function assignment causes error
• Confusing function call syntax

4. Identify the error in this Swift code snippet:

var multiply: (Int, Int) -> Int
multiply = { a, b in a * b }
print(multiply(2))

medium

A. Missing return statement in closure

B. Variable multiply not initialized

C. Calling multiply with wrong number of arguments

D. Incorrect closure syntax

Solution

1. Step 1: Check closure assignment

   The closure { a, b in a * b } correctly matches (Int, Int) -> Int type.

2. Step 2: Check function call

   Calling multiply(2) provides only one argument, but multiply expects two Ints.

3. Final Answer:

   Calling multiply with wrong number of arguments -> Option C

4. Quick Check:

   Argument count mismatch = Calling multiply with wrong number of arguments [OK]

Hint: Match argument count with function type parameters [OK]

Common Mistakes:

• Assuming closure needs explicit return
• Ignoring argument count in function call
• Thinking variable is uninitialized

5. Given the function type (Int) -> (Int) -> Int, what does this type represent in Swift?

hard

A. A function that takes an Int and returns an Int array.

B. A function that takes an Int and returns another function that takes an Int and returns an Int.

C. A function that takes two Ints and returns an Int.

D. A function that takes no parameters and returns an Int.

Solution

1. Step 1: Analyze the nested function type

   The type (Int) -> (Int) -> Int means the function takes one Int and returns another function.

2. Step 2: Understand the returned function

   The returned function itself takes an Int and returns an Int.

3. Final Answer:

   A function that takes an Int and returns another function that takes an Int and returns an Int. -> Option B

4. Quick Check:

   Curried function type = A function that takes an Int and returns another function that takes an Int and returns an Int. [OK]

Hint: Read from left to right: input then output, output can be a function [OK]

Common Mistakes:

• Confusing nested function types with multiple parameters
• Assuming it returns an array
• Ignoring that return type can be a function