找到目标顶点并返回路径时退出 DFS 递归

Question

我做了一个递归 DFS 算法，它可以在找到目标时返回路径。请注意，我正在利用递归，并尝试在路径保存中使用尽可能少的内存。当找到目标时，保证path 将包含正确的路径。但是随后 DFS 必须完全停止，否则由于之后的pop()s，我无法返回正确的列表。临时解决方案是全局列表result。有什么办法：

1. 找到目标后停止递归？
2. 在指定点返回path 列表？

result = []

def dfs(graph, vertex, visited, path):

    global result
    if vertex not in visited:
        visited.add(vertex)
        if isTarget(vertex):
            print("Solution found ! " + str(path))
            result = path.copy() # recursion must stop here and return path at this state
        for neighbor in graph.getSuccessors(vertex):
            path.append(neighbor[1]) #neighbor[1] is direction 
            print("Visiting node" + str(neighbor[0]))
            dfs(graph, neighbor[0], visited, path)
            path.pop()

Answer 1

我建议将搜索逻辑与遍历逻辑分开。这允许您以通用方式编写 dfs 并重用它来遍历您的图形以进行任何与图形相关的操作 -

def dfs(graph, vertex, path = tuple(), visited = set()):
  if vertex in visited: return
  yield (vertex, path)
  for [v,dir] in graph.getSucessors(vertex):
    yield from dfs(graph, v, (*path, dir), {*visited, v})

现在find 只是我们可以围绕dfs 编写的众多函数之一-

def find(graph, start, target):
  for (v, path) in dfs(graph, start):
    if v == target:
      return (v, path)
  return (None, None)

dfs 是惰性的，因此一旦find 返回，迭代就会停止。这是一个如何使用它的示例 -

(vertex, path) = find(graph, startVertex, targetVertex)
if not vertex:
  print("target not found")
else:
  print(f"found {vertex} at path {path}")

这种方法的一个优点是不仅可以找到第一个目标，还可以找到所有个目标 -

def find_all(graph, start, target):
  return list((v, path) for (v, path) in dfs(graph, start) if v == target)

如果您分享graph.getSuccessors，我相信我们也可以对其进行优化。如果你像dfs 那样把它写成懒惰的，那么两个生成器都可以从提前退出行为中受益。

Answer 2

在我看来，dfs() 可以返回一个列表，这将允许我们继续/中断递归。比如：

result = []

def dfs(graph, vertex, visited, path):

    if vertex not in visited:
        visited.add(vertex)
        if isTarget(vertex):
            print("Solution found ! " + str(path))
            return path # recursion stops here and returns path at this state
        for neighbor in graph.getSuccessors(vertex):
            path.append(neighbor[1]) #neighbor[1] is direction 
            print("Visiting node" + str(neighbor[0]))
            temp_path = dfs(graph, neighbor[0], visited, path)
            if temp_path:
                return temp_path # break recursion since path found
            path.pop()
    return [] # if no path/target node found

result = dfs()