python 宽度优先代码

以下是 Python 中宽度优先搜索的代码示例：

from collections import deque

def bfs(graph, start):
    visited = set()
    queue = deque([start])
    while queue:
        vertex = queue.popleft()
        if vertex not in visited:
            visited.add(vertex)
            queue.extend(graph[vertex] - visited)
    return visited

其中，`graph` 是一个字典，表示图的邻接表；`start` 是起始节点。