数据结构深度优先遍历代码
时间: 2024-04-29 19:17:58 浏览: 142
深度优先遍历(Depth First Search,DFS)是一种常用的图遍历算法,也可以用于树的遍历。下面是一个示例的深度优先遍历代码:
```python
# 定义图的类
class Graph:
def __init__(self):
self.graph = {}
# 添加边
def add_edge(self, node, neighbor):
if node in self.graph:
self.graph[node].append(neighbor)
else:
self.graph[node] = [neighbor]
# 深度优先遍历
def dfs(self, start):
visited = set() # 用集合记录已访问的节点
self._dfs(start, visited)
def _dfs(self, node, visited):
visited.add(node)
print(node) # 打印当前节点
neighbors = self.graph.get(node, [])
for neighbor in neighbors:
if neighbor not in visited:
self._dfs(neighbor, visited)
# 创建图对象
graph = Graph()
# 添加边
graph.add_edge('A', 'B')
graph.add_edge('A', 'C')
graph.add_edge('B', 'D')
graph.add_edge('B', 'E')
graph.add_edge('C', 'F')
graph.add_edge('E', 'G')
# 深度优先遍历
graph.dfs('A')
```
上述代码中,首先定义了一个`Graph`类来表示图,其中`add_edge`方法用于添加边,`dfs`方法用于进行深度优先遍历。在深度优先遍历中,使用一个集合`visited`来记录已访问的节点,遍历过程中递归地访问每个节点的邻居节点。
阅读全文