我是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