二叉树的遍历:给定一棵二叉树,实现前序遍历、中序遍历和后序遍历。
时间: 2023-12-18 15:56:38 浏览: 116
二叉树的遍历是指按照一定的顺序遍历二叉树的所有节点。常用的遍历方式有前序遍历、中序遍历和后序遍历。
前序遍历:先访问根节点,再访问左子树,最后访问右子树。即根->左->右的顺序。
中序遍历:先访问左子树,再访问根节点,最后访问右子树。即左->根->右的顺序。
后序遍历:先访问左子树,再访问右子树,最后访问根节点。即左->右->根的顺序。
下面是二叉树遍历的代码实现:
```python
# 定义二叉树节点
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
# 前序遍历
def preorderTraversal(root):
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
if node:
res.append(node.val)
stack.append(node.right)
stack.append(node.left)
return res
# 中序遍历
def inorderTraversal(root):
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):
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
if node:
res.append(node.val)
stack.append(node.left)
stack.append(node.right)
return res[::-1]
```
以上代码中,我们使用栈来实现遍历。在前序遍历和后序遍历中,我们采用先访问右子树再访问左子树的方式,这样才能保证遍历的顺序是根->左->右或者左->右->根。在中序遍历中,我们需要不断将左子树节点入栈,直到左子树为空为止。然后弹出栈顶节点,再将右子树节点入栈。这样可以保证遍历的顺序是左->根->右。
阅读全文