forked from TheAlgorithms/Python
- Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcount_paths.py
75 lines (62 loc) · 2.1 KB
/
count_paths.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
"""
Given a grid, where you start from the top left position [0, 0],
you want to find how many paths you can take to get to the bottom right position.
start here -> 0 0 0 0
1 1 0 0
0 0 0 1
0 1 0 0 <- finish here
how many 'distinct' paths can you take to get to the finish?
Using a recursive depth-first search algorithm below, you are able to
find the number of distinct unique paths (count).
'*' will demonstrate a path
In the example above, there are two distinct paths:
1. 2.
* * * 0 * * * *
1 1 * 0 1 1 * *
0 0 * 1 0 0 * 1
0 1 * * 0 1 * *
"""
defdepth_first_search(grid: list[list[int]], row: int, col: int, visit: set) ->int:
"""
Recursive Backtracking Depth First Search Algorithm
Starting from top left of a matrix, count the number of
paths that can reach the bottom right of a matrix.
1 represents a block (inaccessible)
0 represents a valid space (accessible)
0 0 0 0
1 1 0 0
0 0 0 1
0 1 0 0
>>> grid = [[0, 0, 0, 0], [1, 1, 0, 0], [0, 0, 0, 1], [0, 1, 0, 0]]
>>> depth_first_search(grid, 0, 0, set())
2
0 0 0 0 0
0 1 1 1 0
0 1 1 1 0
0 0 0 0 0
>>> grid = [[0, 0, 0, 0, 0], [0, 1, 1, 1, 0], [0, 1, 1, 1, 0], [0, 0, 0, 0, 0]]
>>> depth_first_search(grid, 0, 0, set())
2
"""
row_length, col_length=len(grid), len(grid[0])
if (
min(row, col) <0
orrow==row_length
orcol==col_length
or (row, col) invisit
orgrid[row][col] ==1
):
return0
ifrow==row_length-1andcol==col_length-1:
return1
visit.add((row, col))
count=0
count+=depth_first_search(grid, row+1, col, visit)
count+=depth_first_search(grid, row-1, col, visit)
count+=depth_first_search(grid, row, col+1, visit)
count+=depth_first_search(grid, row, col-1, visit)
visit.remove((row, col))
returncount
if__name__=="__main__":
importdoctest
doctest.testmod()