python 二叉树遍历
时间: 2023-11-05 17:00:35 浏览: 96
python二叉树遍历的实现方法
二叉树遍历是指按照一定的规则访问二叉树中的每一个节点。常见的二叉树遍历有前序遍历、中序遍历和后序遍历。
1. 前序遍历:按照根节点->左子树->右子树的顺序遍历二叉树。
2. 中序遍历:按照左子树->根节点->右子树的顺序遍历二叉树。
3. 后序遍历:按照左子树->右子树->根节点的顺序遍历二叉树。
下面是Python代码实现:
```python
# 定义二叉树节点
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
# 前序遍历
def preorderTraversal(root: TreeNode) -> List[int]:
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
res.append(node.val)
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return res
# 中序遍历
def inorderTraversal(root: TreeNode) -> List[int]:
if not root:
return []
res = []
stack = []
while stack or root:
while root:
stack.append(root)
root = root.left
node = stack.pop()
res.append(node.val)
root = node.right
return res
# 后序遍历
def postorderTraversal(root: TreeNode) -> List[int]:
if not root:
return []
res = []
stack = []
while stack or root:
while root:
stack.append(root)
root = root.left if root.left else root.right
node = stack.pop()
res.append(node.val)
if stack and stack[-1].left == node:
root = stack[-1].right
return res
```
其中,preorderTraversal函数实现前序遍历,inorderTraversal函数实现中序遍历,postorderTraversal函数实现后序遍历。遍历过程中使用了栈来辅助实现。
阅读全文