在图形中查找顶点的邻居-修改后的bfs

时间:2018-11-04 12:26:39

标签: python graph-theory breadth-first-search

我是python的初学者,所以我将非常感谢您的帮助。 :)

我需要编写一个函数,使用修改的bfs返回给定距离内给定顶点的所有邻居。我的代码在下面,但我不知道如何更正。

def bfs_neighbors(graph: List, first_vertex: int, distance: int) -> Set:
    visited, queue = set(), [first_vertex]
    for i in range(distance):
        while queue:
            vertex = queue.pop(0)
            if vertex not in queue:
                visited.add(vertex)
                if not visited:
                    queue.extend(graph[vertex])
        return visited

0 个答案:

没有答案