contents: add bfs algorithm (#137)

* Add new Graph traversal implementation

* Add seperate BFS implementation

* Cleanup code and comments

* Rename variable and remove another comment

* Switch to deque for DFS implementation

Signed-off-by: Aadit Kamat <aadit.k12@gmail.com>

* Switch back to recursive DFS and check for empty graph

Signed-off-by: Aadit Kamat <aadit.k12@gmail.com>

* Update graph.md
pull/141/head
Aadit Kamat 5 years ago committed by Yangshun Tay
parent ed14d9ce5d
commit 473dfb2d22

@ -27,29 +27,73 @@ A tree-like diagram could very well be a graph that allows for cycles and a naiv
- **Uncommon** - Topological Sort, Dijkstra's algorithm - **Uncommon** - Topological Sort, Dijkstra's algorithm
- **Rare** - Bellman-Ford algorithm, Floyd-Warshall algorithm, Prim's algorithm, Kruskal's algorithm - **Rare** - Bellman-Ford algorithm, Floyd-Warshall algorithm, Prim's algorithm, Kruskal's algorithm
In coding interviews, graphs are commonly represented as 2-D matrices where cells are the nodes and each cell can traverse to its adjacent cells (up/down/left/right). Hence it is important that you be familiar with traversing a 2-D matrix. When recursively traversing the matrix, always ensure that your next position is within the boundary of the matrix. More tips for doing depth-first searches on a matrix can be found [here](https://discuss.leetcode.com/topic/66065/python-dfs-bests-85-tips-for-all-dfs-in-matrix-question/). A simple template for doing depth-first searches on a matrix goes like this: In coding interviews, graphs are commonly represented as 2-D matrices where cells are the nodes and each cell can traverse to its adjacent cells (up/down/left/right). Hence it is important that you be familiar with traversing a 2-D matrix. When traversing the matrix, always ensure that your current position is within the boundary of the matrix and has not been visited before.
A simple template for doing depth-first searches on a matrix goes like this:
```py ```py
def traverse(matrix): def dfs(matrix):
# Check for an empty graph.
if not matrix:
return []
rows, cols = len(matrix), len(matrix[0]) rows, cols = len(matrix), len(matrix[0])
visited = set() visited = set()
directions = ((0, 1), (0, -1), (1, 0), (-1, 0)) directions = ((0, 1), (0, -1), (1, 0), (-1, 0))
def dfs(i, j):
def traverse(i, j):
if (i, j) in visited: if (i, j) in visited:
return return
visited.add((i, j)) visited.add((i, j))
# Traverse neighbors # Traverse neighbors.
for direction in directions: for direction in directions:
next_i, next_j = i + direction[0], j + direction[1] next_i, next_j = i + direction[0], j + direction[1]
if 0 <= next_i < rows and 0 <= next_j < cols: # Check boundary if 0 <= next_i < rows and 0 <= next_j < cols:
# Add any other checking here ^ # Add in your question-specific checks.
dfs(next_i, next_j) traverse(next_i, next_j)
for i in range(rows):
for j in range(cols):
traverse(i, j)
```
A similar template for doing breadth-first searches on the matrix goes like this:
```py
from collections import deque
def bfs(matrix):
# Check for an empty graph.
if not matrix:
return []
rows, cols = len(matrix), len(matrix[0])
visited = set()
directions = ((0, 1), (0, -1), (1, 0), (-1, 0))
def traverse(i, j):
queue = deque([(i, j)])
while queue:
curr_i, curr_j = queue.pop()
if (curr_i, curr_j) not in visited:
visited.add((curr_i, curr_j))
# Traverse neighbors.
for direction in directions:
next_i, next_j = curr_i + direction[0], curr_j + direction[1]
if 0 <= next_i < rows and 0 <= next_j < cols:
# Add in your question-specific checks.
queue.append((next_i, next_j))
for i in range(rows): for i in range(rows):
for j in range(cols): for j in range(cols):
dfs(i, j) traverse(i, j)
``` ```
> NOTE: While DFS is implemented using recursion in this sample, it could also be implemented iteratively similar to BFS. The key difference between the algorithms lies in the underlying data structure (BFS uses a queue while DFS uses a stack). The `deque` class in Python can function as both a stack and a queue
For additional tips on BFS and DFS, you can refer to this [LeetCode post](https://leetcode.com/problems/pacific-atlantic-water-flow/discuss/90774/Python-solution-with-detailed-explanation)
## Corner cases ## Corner cases
- Empty graph - Empty graph

Loading…
Cancel
Save