python遍历二叉树注意事项
时间: 2023-12-19 11:32:31 浏览: 28
遍历二叉树时需要注意以下几点:
1. 确定遍历的顺序:二叉树的遍历有三种方式,分别是前序遍历、中序遍历和后序遍历。前序遍历是先访问根节点,然后递归地遍历左子树和右子树;中序遍历是先递归地遍历左子树,然后访问根节点,最后递归地遍历右子树;后序遍历是先递归地遍历左子树和右子树,最后访问根节点。
2. 确定遍历的终止条件:在递归遍历二叉树时,需要确定递归的终止条件。一般情况下,当节点为空时,递归结束。
3. 处理节点的操作:在遍历二叉树的过程中,可以对每个节点进行相应的操作,例如打印节点的值、将节点的值存入列表等。
以下是一个示例代码,演示了如何遍历二叉树:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
def inorderTraversal(root):
if root is None:
return []
result = []
stack = []
while root or stack:
while root:
stack.append(root)
root = root.left
root = stack.pop()
result.append(root.val)
root = root.right
return result
# 创建二叉树
root = TreeNode(2)
root.left = TreeNode(1)
root.right = TreeNode(3)
# 中序遍历二叉树
result = inorderTraversal(root)
print(result) # 输出:[1, 2, 3]
```