Chapters

Hide chapters

Data Structures & Algorithms in Swift

Fourth Edition · iOS 15 · Swift 5.5 · Xcode 13

38. Breadth-First Search
Written by Vincent Ngo

Heads up... You're reading this book for free, with parts of this chapter shown beyond this point as scrambled text.

In the previous chapter, you explored using graphs to capture relationships between objects. Remember that objects are just vertices, and edges represent the relationships between them.

Several algorithms exist to traverse or search through a graph’s vertices. One such algorithm is the breadth-first search (BFS) algorithm.

BFS can be used to solve a wide variety of problems:

  1. Generating a minimum-spanning tree.
  2. Finding potential paths between vertices.
  3. Finding the shortest path between two vertices.

Example

BFS starts by selecting any vertex in a graph. The algorithm then explores all neighbors of this vertex before traversing the neighbors of said neighbors and so forth. As the name suggests, this algorithm takes a breadth-first approach.

Going through a BFS example using the following undirected graph:

H E D F G B A C

Note: Highlighted vertices represent visited vertices.

You will use a queue to keep track of which vertices to visit next. The first-in-first-out approach of the queue guarantees that all of a vertex’s neighbors are visited before you traverse one level deeper.

D C H G F E B 1 A Queue H G F E 2 A B D C Queue A A B C D

  1. To begin, you pick a source vertex to start from. Here, you have chosen A, which is added to the queue.
  2. As long as the queue is not empty, you dequeue and visit the next vertex, in this case, A. Next, you add all of A’s neighboring vertices [B, D, C] to the queue.

Note: It’s important to note that you only add a vertex to the queue when it has not yet been visited and is not already in the queue.

A B D C E Queue B A D C E Queue H G F 4 A B D C E H G F 3 A B D C E

  1. The queue is not empty, so you dequeue and visit the next vertex, B. You then add B’s neighbor E to the queue. A is already visited, so it does not get added. The queue now has [D, C, E].

  2. The next vertex to be dequeued is D. D does not have any neighbors that aren’t visited. The queue now has [C, E].

A C B E D F G H Queue B C A D E F G Queue 6 A B E C D F H H 5 A D C B E F G G

  1. Next, you dequeue C and add its neighbors [F, G] to the queue. The queue now has [E, F, G].

Note that you have now visited all of A’s neighbors! BFS now moves on to the second level of neighbors.

  1. You dequeue E and add H to the queue. The queue now has [F, G, H]. You don’t add B or F to the queue because B is already visited and F is already in the queue.

A C B E F G D H Queue B C E F A D G H Queue 8 A B E C F G D H 7 A D C F B E G H

  1. You dequeue F, and since all its neighbors are already in the queue or visited, you don’t add anything to the queue.

  2. Like the previous step, you dequeue G and don’t add anything to the queue.

B C E F G H A D Queue 10 9 A D C G H F B E A B D C H E F G

  1. Finally, you dequeue H. The breadth-first search is complete since the queue is now empty!

  2. When exploring the vertices, you can construct a tree-like structure, showing the vertices at each level: first the vertex you started from, then its neighbors, then its neighbors’ neighbors and so on.

Implementation

Open up the starter playground for this chapter. This playground contains an implementation of a graph you built in the previous chapter. It also includes a stack-based queue implementation, which you will use to implement BFS.

extension Graph where Element: Hashable {

  func breadthFirstSearch(from source: Vertex<Element>)
      -> [Vertex<Element>] {
    var queue = QueueStack<Vertex<Element>>()
    var enqueued: Set<Vertex<Element>> = []
    var visited: [Vertex<Element>] = []

    // more to come

    return visited
  }
}
queue.enqueue(source) // 1
enqueued.insert(source)

while let vertex = queue.dequeue() { // 2
  visited.append(vertex) // 3
  let neighborEdges = edges(from: vertex) // 4
  neighborEdges.forEach { edge in
    if !enqueued.contains(edge.destination) { // 5
      queue.enqueue(edge.destination)
      enqueued.insert(edge.destination)
    }
  }
}
let vertices = graph.breadthFirstSearch(from: a)
vertices.forEach { vertex in
  print(vertex)
}
0: A
1: B
2: C
3: D
4: E
5: F
6: G
7: H

Performance

When traversing a graph using BFS, each vertex is enqueued once. This process has a time complexity of O(V). During this traversal, you also visit all the edges. The time it takes to visit all edges is O(E). Adding the two together means that the overall time complexity for breadth-first search is O(V + E).

Key points

  • Breadth-first search (BFS) is an algorithm for traversing or searching a graph.
  • BFS explores all the current vertex’s neighbors before traversing the next level of vertices.
  • It’s generally good to use this algorithm when your graph structure has many neighboring vertices or when you need to find out every possible outcome.
  • The queue data structure is used to prioritize traversing a vertex’s edges before diving down a level deeper.
Have a technical question? Want to report a bug? You can ask questions and report bugs to the book authors in our official book forum here.
© 2024 Kodeco Inc.

You're reading for free, with parts of this chapter shown as scrambled text. Unlock this book, and our entire catalogue of books and videos, with a Kodeco Personal Plan.

Unlock now