深度优先搜索Python实现
时间: 2024-04-20 11:21:01 浏览: 110
深度优先搜索(Depth-First Search,DFS)是一种常用的图遍历算法,它通过递归的方式从起始节点开始,沿着一条路径尽可能深入地搜索,直到无法继续为止,然后回溯到上一个节点,继续搜索其他路径。下面是深度优先搜索的Python实现:
```python
# 定义图的类
class Graph:
def __init__(self):
self.graph = {}
# 添加边
def add_edge(self, u, v):
if u in self.graph:
self.graph[u].append(v)
else:
self.graph[u] = [v]
# 深度优先搜索
def dfs(self, start):
visited = set() # 记录已访问的节点
self._dfs(start, visited)
def _dfs(self, node, visited):
visited.add(node)
print(node, end=" ") # 输出当前节点
neighbors = self.graph.get(node, [])
for neighbor in neighbors:
if neighbor not in visited:
self._dfs(neighbor, visited)
# 创建图对象
g = Graph()
# 添加边
g.add_edge(1, 2)
g.add_edge(1, 3)
g.add_edge(2, 4)
g.add_edge(2, 5)
g.add_edge(3, 6)
g.add_edge(3, 7)
# 从节点1开始进行深度优先搜索
print("深度优先搜索结果:")
g.dfs(1)
```
以上代码中,首先定义了一个`Graph`类来表示图,其中`add_edge`方法用于添加边,`dfs`方法用于进行深度优先搜索。在`_dfs`方法中,使用递归的方式进行深度优先搜索,并通过一个`visited`集合来记录已访问的节点,避免重复访问。
阅读全文