从连通无向图或强连通有向图中任一顶点出发遍历,或从有根有向图的根顶点出发遍历,都可以访问到所有顶点。在遍历中经过的边加上原图的所有顶点,就构成该图的一棵生成树。通过遍历构造生成树的过程可以按深度优先方式或宽度优先方式进行,在遍历中记录访问的所有顶点和经过的边,就得到原图的深度优先生成树(简称DFS生成树),或者宽度优先生成树(BFS生成树)。
在得到的生成树中,每个节点可能有多个后一节点,但都有唯一的前一顶点,可以采用存储该节点前一节点以及边的方式来表示一棵生成树。即span_forest[i]=(j,e)表示i节点的上一个节点是j,边为e。特别地,如果该节点为根节点,那么它的前一节点为它本身,边为0。
采用递归方式构造DFS生成树的代码如下。
def DFS_span_forest(graph):
vnum=graph.vertex_num()
span_forest=[None]*vnum
def dfs(graph,v): #递归遍历函数,在递归中记录经由边
nonlocal span_forest#span_forest是nonlocal变量,即始终引用外层的span_forest
for u,w in graph.out_edges(v):
if span_forest[u] is None:
span_forest[u]=[v,w]
dfs(graph,u)
for v in range(vnum):
if span_forest[v] is None:
span_forest[v]=(v,0)
dfs(graph,v)
return span_forest