forked from HarshCasper/NeoAlgo
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path2D_DFS_Recursive.py
161 lines (125 loc) · 4.7 KB
/
2D_DFS_Recursive.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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
"""
Depth First Search: Its a graph traversal technique which uses
stack data-structure fo its implementation. We extract the
current node from the top of the stack and traverse its adjacent
nodes. If not visited, we put the node at the top of the stack
mark it as visited.
Purpose: Given a binary matrix of N*M order where 0 is the wall and 1 is the way.
Find the shortest distance from a source cell to a destination cell,
traversing through limited cells only. Also, you can move only
up, down, left and right. If found then print the distance and
path in separate lines, else return -1.
Method : Depth-First Search
Intuition : As we know stack is used for DFS. In recursion
1. Initialize stack.
2. Initialize 2d boolean array, the same size as the original array.
This will help us in avoiding traversal to go in loops.
3. Make a call to recursive function DFSUtil with row=0, column =0
and visited[] array.
4. In DFSutil Function, check if row and column indexes are within
the range of given matrix and marked false in the visited[] array,
if not then return from the function. If indexes are valid and not
visited then print the element. Mark the element in the visited array.
Make a recursive call to left(column-1), right(column+1), down(row+1)
and up(row+1) from the current element position.
Time Complexity: O(N*M)
Space Complexity: O(N*M)
Argument: 2-d list, tuple, tuple (Maze, Source, Destination)
Return : Integer, String (Distance, Path)
"""
# Main DFS function using recursion
def DFS(maze, Dim, cur_pos, des, visited, path, moves):
x, y = cur_pos
n, m = Dim
# Base case: if the current position is the destination, return the path
if cur_pos == des:
return path
# Check for each possible move from a possition
for i in moves.keys():
r = x + i[0]
c = y + i[1]
# If the next node inside the maze , has a way and not yet visited
# then mark it visited, add the path and recursively call DFS()
# Function from that position
if 0 <= r < n and 0 <= c < m and maze[r][c] == 1 and not visited[r][c]:
visited[r][c] = True
path += moves[i]
temp = DFS(maze, Dim, (r, c), des, visited, path, moves)
# If the path is found, simply return path
if temp:
return temp
# If the path is not found, pop the path
path.pop()
# If no path from the current node is possible, return False
return False
def Find_Path(maze, src, des):
# Base Case: If there is no way from the source, returnn False
if(maze[src[0]][src[1]] != 1):
return False
# Dimention of the maze
n = len(maze)
m = len(maze[0])
# To keep a track of visited nodes, also mark source as visited
visited = [[False] * m for i in range(n)]
visited[src[0]][src[1]] = True
# All possible moves from a cell
moves = {(1, 0): 'D', (-1, 0): 'U', (0, 1): 'R', (0, -1): 'L'}
path = DFS(maze, (n, m), src, des, visited, [], moves)
# if the path is not found, return False
if path is False:
return False
return len(path), "".join(path)
# --------------------------------DRIVER CODE ---------------------------------
if __name__ == "__main__":
N, M = map(int, input("Enter the Dimension of the maze:- ").split())
print("Enter the Maze: ")
maze = []
# Input the Maze
for _ in range(N):
maze.append([int(i) for i in input().split()])
src = tuple(map(int, input("Enter the Source cell: ").split()))
des = tuple(map(int, input("Enter the Destination cell: ").split()))
ans = Find_Path(maze, src, des)
# If ans is false, i.e. no way is possible, else print distance and path
if ans is False:
print("No Path exists between", src, "and", des)
else:
dist = ans[0]
path = ans[1]
print("Disance= ", dist)
print("Path: ", path)
"""
Sample Input / Output
Enter the Dimension of the maze:- 5 5
Enter the Maze:
1 0 1 1 1
1 0 1 0 1
1 0 1 0 1
1 0 1 0 1
1 1 1 0 1
Enter the Source cell: 0 0
Enter the Destination cell: 4 4
Disance= 16
Path: DDDDRRUUUURRDDDD
Enter the Dimension of the maze:- 5 5
Enter the Maze:
1 0 1 1 1
1 0 1 0 1
1 0 0 0 1
1 0 1 0 1
1 1 1 0 1
Enter the Source cell: 0 0
Enter the Destination cell: 4 4
No Path exists between (0, 0) and (4, 4)
Enter the Dimension of the maze:- 5 8
Enter the Maze:
1 0 1 1 1 1 1 1
1 0 1 0 0 0 0 1
1 0 1 1 1 1 0 1
1 0 0 0 0 1 0 1
1 1 1 1 1 1 0 1
Enter the Source cell: 0 0
Enter the Destination cell: 4 7
Disance= 25
Path: DDDDRRRRRUULLLUURRRRRDDDD
"""