Tracing and Returning a Path in Depth First Search Tracing and Returning a Path in Depth First Search python python

Tracing and Returning a Path in Depth First Search


You are right - you cannot simply return the stack, it indeed contains a lot of unvisited nodes.

However, by maintaining a map (dictionary): map:Vertex->Vertex such that parentMap[v] = the vertex we used to discover v, you can get your path.

The modification you will need to do is pretty much in the for loop:

    for child in children:        stack.push(child[0])        parentMap[child] = parent #this line was added

Later on, when you found your target, you can get the path from the source to the target (pseudo code):

curr = targetwhile (curr != None):  print curr  curr = parentMap[curr]

Note that the order will be reversed, it can be solved by pushing all elements to a stack and then print.

I once answered a similar (though not identical IMO) question regarding finding the actual path in BFS in this thread

Another solution is to use a recursive version of DFS rather then iterative+stack, and once a target is found, print all current nodes in the recursion back up - but this solution requires a redesign of the algorithm to a recursive one.


P.S. Note that DFS might fail to find a path to the target (even if maintaining a visited set) if the graph contains an infinite branch.
If you want a complete (always finds a solution if one exists) and optimal (finds shortest path) algorithm - you might want to use BFS or Iterative Deepening DFS or even A* Algorithm if you have some heuristic function


Not specific to your problem, but you can tweak this code and apply it to different scenarios, in fact, you can make the stack also hold the path.

Example:

     A   /    \  C      B  \     / \   \    D E    \    /       F       
graph = {'A': set(['B', 'C']),         'B': set(['A', 'D', 'E']),         'C': set(['A', 'F']),         'D': set(['B']),         'E': set(['B', 'F']),         'F': set(['C', 'E'])}def dfs_paths(graph, start, goal):    stack = [(start, [start])]    visited = set()    while stack:        (vertex, path) = stack.pop()        if vertex not in visited:            if vertex == goal:                return path            visited.add(vertex)            for neighbor in graph[vertex]:                stack.append((neighbor, path + [neighbor]))print (dfs_paths(graph, 'A', 'F'))   #['A', 'B', 'E', 'F']


this link should help you alot ... It is a lengthy article that talks extensively about a DFS search that returns a path... and I feel it is better than any answer I or anyone else can post

http://www.python.org/doc/essays/graphs/