FlatMap for nested collections in Swift - Step-by-Step Execution

Concept Flow - FlatMap for nested collections

Start with nested collection

↓

Apply flatMap

↓

Extract inner collections

↓

Flatten into single collection

↓

Result: one-level collection

FlatMap takes a nested collection and flattens it into a single-level collection by extracting and combining inner collections.

Execution Sample

Swift

let nested = [[1, 2], [3, 4], [5]]
let flat = nested.flatMap { $0 }
print(flat)

This code flattens a nested array of integers into a single array.

Execution Table

Step	Action	Input	Output
1	Start with nested array	[[1, 2], [3, 4], [5]]	[[1, 2], [3, 4], [5]]
2	Apply flatMap to first inner array	[1, 2]	[1, 2]
3	Apply flatMap to second inner array	[3, 4]	[1, 2, 3, 4]
4	Apply flatMap to third inner array	[5]	[1, 2, 3, 4, 5]
5	Final flattened array	N/A	[1, 2, 3, 4, 5]

💡 All inner arrays processed and combined into one flat array.

Variable Tracker

Variable	Start	After 1	After 2	After 3	Final
nested	[[1, 2], [3, 4], [5]]	[[1, 2], [3, 4], [5]]	[[1, 2], [3, 4], [5]]	[[1, 2], [3, 4], [5]]	[[1, 2], [3, 4], [5]]
flat	[]	[1, 2]	[1, 2, 3, 4]	[1, 2, 3, 4, 5]	[1, 2, 3, 4, 5]

Key Moments - 3 Insights

Why does flatMap use $0 inside the closure?

Because $0 represents each inner array inside the nested array, flatMap extracts these inner arrays to combine them into one flat array, as shown in execution_table rows 2-4.

Is flatMap the same as map followed by flatten?

Yes, flatMap combines mapping and flattening in one step. Here, flatMap { $0 } is like mapping each inner array to itself and then flattening, as seen in the stepwise output.

What happens if the nested array is empty?

If nested is empty, flatMap returns an empty array immediately, as no inner arrays exist to process, similar to the start state in execution_table row 1.

Visual Quiz - 3 Questions

Look at the execution_table, what is the value of 'flat' after processing the second inner array?

A [1, 2, 3, 4]

B [3, 4]

C [1, 2]

D [1, 2, 3]

💡 Hint

Check execution_table row 3 under Output column.

At which step does the flat array contain all elements from the nested arrays?

A Step 2

B Step 5

C Step 3

D Step 4

💡 Hint

Look at execution_table row 5 Output column for the final flat array.

If the nested array was [[1], [], [2, 3]], what would flatMap output after processing the second inner array?

A [1, ]

B [1, 2, 3]

C [1]

D ] ,1[

💡 Hint

Empty inner arrays add no elements; check how flat changes after each inner array in variable_tracker.

Concept Snapshot

flatMap for nested collections:
- Use flatMap to flatten nested arrays
- Syntax: nested.flatMap { $0 }
- Extracts and combines inner arrays
- Result is a single-level array
- Useful to avoid nested loops

Full Transcript

This visual execution shows how flatMap works on nested collections in Swift. Starting with a nested array of arrays, flatMap takes each inner array ($0) and combines them into one flat array. Step by step, each inner array is appended to the result. Variables 'nested' and 'flat' track the original and flattened arrays. Key moments clarify why $0 is used and how flatMap differs from map. The quiz tests understanding of the flattening process and final output.