pythonfrom collections import dequedef bfs_traverse(graph, start): """ 对无权无向图进行BFS遍历 :param graph: 图的邻接表表示,如 {0: [1,2], 1: [0,3], ...} :param start: 起始节点 :return: 遍历顺序列表 """ visited = set() # 记录已访问节点 queue = deque([start]) # 初始化队列,加入起点 visited.add(start) traversal_order = [] # 存储遍历结果 while queue: node = queue.popleft() # 取出队首节点 traversal_order.append(node) # 遍历当前节点的所有邻居 for neighbor in graph[node]: if neighbor not in visited: visited.add(neighbor) queue.append(neighbor) return traversal_order# 测试:创建一个简单图(邻接表)if __name__ == "__main__": # 图结构:0-1-2-3,且0-2相连(形成环) graph = { 0: [1, 2], 1: [0, 3], 2: [0, 3], 3: [1, 2] } result = bfs_traverse(graph, 0) print("BFS遍历顺序:", result) # 输出: [0, 1, 2, 3]运行结果解析:从节点0出发,先访问其相邻的1和2(第一层),然后从1和2分别访问3(第二层)。由于3被先加入队列的1访问到,所以顺序为0→1→2→3。### 关键点注释-collections.deque提供了O(1)的左右两端操作,非常适合BFS。-visited使用集合,查找时间复杂度为O(1)。- 每一步都保证了节点的最早访问,从而确保最短路径性质。## BFS的应用场景与变体### 1. 最短路径问题(无权图)BFS的一个经典应用是在无权图中寻找从起点到终点的最短路径。只需在访问节点时记录其前驱节点,最后逆向回溯即可得到路径。### 2. 连通分量检测在社交网络或网格图中,BFS可以快速找出所有连通的组件。例如,遍历所有节点,每启动一次BFS就发现一个连通分量。### 3. 迷宫求解在二维网格中,BFS可以找到从入口到出口的最短路径,每一步的代价相同(如1步)。下面是一个具体例子。## 代码示例二:用BFS求解迷宫最短路径假设有一个二维迷宫,用0表示可行走区域,1表示墙壁,起点为(0,0),终点为(rows-1, cols-1)。我们需要找出最短路径长度。pythonfrom collections import dequedef bfs_maze(maze, start, end): """ 在迷宫中寻找最短路径长度(BFS) :param maze: 二维列表,0表示路,1表示墙 :param start: 起点坐标 (r, c) :param end: 终点坐标 (r, c) :return: 最短路径步数,若不可达则返回-1 """ rows, cols = len(maze), len(maze[0]) # 四个方向:上、下、左、右 directions = [(-1, 0), (1, 0), (0, -1), (0, 1)] # visited 记录已访问坐标,避免重复 visited = [[False] * cols for _ in range(rows)] queue = deque([(start[0], start[1], 0)]) # (行, 列, 步数) visited[start[0]][start[1]] = True while queue: r, c, steps = queue.popleft() # 到达终点 if (r, c) == end: return steps # 探索四个方向 for dr, dc in directions: nr, nc = r + dr, c + dc # 检查边界、墙壁、是否已访问 if 0 <= nr < rows and 0 <= nc < cols and maze[nr][nc] == 0 and not visited[nr][nc]: visited[nr][nc] = True queue.append((nr, nc, steps + 1)) return -1 # 无法到达# 测试迷宫(5x5,0表示路,1表示墙)if __name__ == "__main__": maze = [ [0, 0, 1, 0, 0], [0, 0, 0, 0, 1], [1, 1, 0, 1, 0], [0, 0, 0, 0, 0], [0, 1, 1, 0, 0] ] start = (0, 0) end = (4, 4) steps = bfs_maze(maze, start, end) print(f"从{start}到{end}的最短步数: {steps}") # 输出: 8代码解析: - 队列中存储三元组(r, c, steps),其中 steps 记录从起点到当前点的步数。- 每次扩展时,步数加1,直接体现了BFS的层级特性。- 当首次遇到终点时,步数即为最短路径长度,因为BFS保证按层级递增访问。## 复杂度分析### 时间复杂度- 对于图:O(V + E),其中V是节点数,E是边数。每个节点入队一次,每条边被检查一次(无向图每条边被两个节点各检查一次,但整体仍为O(E))。- 对于网格:O(rows * cols),因为每个格子最多被访问一次。### 空间复杂度- 最坏情况:O(V),队列中可能同时存储所有节点(如完全图)。在网格中,空间复杂度为O(rows * cols)。## BFS与DFS的对比| 特性 | BFS | DFS(深度优先搜索) ||------|-----|-------------------|| 数据结构 | 队列 | 栈(递归或显式) || 搜索策略 | 先广后深 | 先深后广 || 最短路径 | 能找到无权图的最短路径 | 不能保证(除非遍历全部) || 空间消耗 | 通常更大(存储宽层) | 通常更小(存储单条路径) || 适用场景 | 最短路径、层次遍历 | 拓扑排序、连通性检测、回溯 |## 总结BFS是一种基础但极其强大的搜索算法。它的核心在于利用队列实现“层层推进”的机制,从而在无权图中保证找到最短路径。理解BFS需要掌握三个关键点:队列的使用、访问标记的维护、层级的记录。通过本文的两个代码示例(图遍历和迷宫求解),你应该能直观感受到BFS的运作过程。在实际应用中,BFS广泛应用于网络爬虫、社交网络分析、GPS导航、人工智能中的状态空间搜索等场景。掌握BFS不仅是学习算法的基础,更是解决复杂问题的利器。希望本文能帮助你深入理解这一经典算法的原理与实现,并在实践中灵活运用。