二叉树的前中后序遍历实现
时间: 2023-11-03 17:15:20 浏览: 43
二叉树的前、中、后序遍历是二叉树的三种基本遍历方式,它们的实现方式如下:
1. 前序遍历:
前序遍历的访问顺序是:根节点 -> 左子树 -> 右子树。实现方式如下:
```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
```
2. 中序遍历:
中序遍历的访问顺序是:左子树 -> 根节点 -> 右子树。实现方式如下:
```python
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
```
3. 后序遍历:
后序遍历的访问顺序是:左子树 -> 右子树 -> 根节点。实现方式如下:
```python
def postorderTraversal(root: TreeNode) -> List[int]:
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
res.append(node.val)
return res[::-1]
```
以上代码中,使用了栈来实现二叉树的遍历。对于前序遍历和后序遍历,需要用到一个辅助栈来记录节点的访问顺序。对于中序遍历,也可以用栈来实现,每次将节点的左子树都入栈,直到左子树为空,再弹出栈顶元素,并将右子树入栈。