为了记录搜索的轨迹,广度优先搜索将每个顶点都找色为白色、灰色和黑色。算法开始时,所有的顶点都是白色的。伴随着搜索的进行,各顶点会逐渐变成灰色,然后成为黑色。与黑色顶点相邻的所有定点都是已经被发现的。灰色顶点可能会有一些白色的相邻顶点,他们代表了已发现与未发现顶点之间的边界。
对于每一个顶点u,其色彩存储于变量color[u]中,u的父母存于变量pai[u]中。
具体代码如下所示(python实现)
def GFS(G, s): print s color = {} d = {} pai = {} V = G.keys() V.remove(s) for u in V: color[u] = 'WHITE' d[u] = '∞' pai[u] = 'NIL' d[s] = 0 pai[s] = 'NIL' Q = [] Q.append(s) while Q != []: u = Q.pop(); for v in G[u]: if color[v] == 'WHITE': color[v] = 'GRAY' print v d[v] = d[u] + 1 pai[v] = u Q.append(v) color[u] = 'BLACK' G ={'A':['B','C'], 'B':['D'], 'C':['E'], 'D':['C','E'], 'E':[]} GFS(G,'A')