python实现二叉树的广度优先遍历(BFS)和深度优先遍历(DFS)
时间: 2023-10-30 22:07:31 浏览: 142
二叉树的广度优先遍历(BFS)可以使用队列来实现,具体步骤如下:
1. 创建一个空队列,并将根节点入队。
2. 当队列不为空时,执行以下操作:
- 出队一个节点,将其值存储或打印。
- 将出队节点的左子节点入队(如果存在)。
- 将出队节点的右子节点入队(如果存在)。
3. 重复步骤2,直到队列为空。
下面是一个Python实现的示例代码:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def bfs(root):
if not root:
return []
result = []
queue = [root]
while queue:
node = queue.pop(0)
result.append(node.val)
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
return result
```
对于深度优先遍历(DFS),有两种常用的方法:前序遍历、中序遍历和后序遍历。下面以前序遍历为例进行解释。
前序遍历的顺序是:根节点 -> 左子树 -> 右子树。具体实现步骤如下:
1. 创建一个空列表,用于存储遍历结果。
2. 定义一个辅助函数,接收一个节点作为参数:
- 若节点为空,返回。
- 将节点的值存储或打印。
- 递归调用辅助函数遍历节点的左子树。
- 递归调用辅助函数遍历节点的右子树。
3. 调用辅助函数,将根节点作为参数传入。
下面是一个Python实现的示例代码:
```python
def dfs(root):
if not root:
return []
result = []
def helper(node):
if not node:
return
result.append(node.val)
helper(node.left)
helper(node.right)
helper(root)
return result
```
这样,你就可以使用这两个函数来实现二叉树的广度优先遍历和深度优先遍历了。
阅读全文