写一段二叉树的遍历代码
时间: 2023-12-01 22:04:32 浏览: 93
二叉树遍历源码
二叉树的遍历有三种方式:前序遍历、中序遍历和后序遍历。
前序遍历的顺序是:根节点 -> 左子树 -> 右子树;
中序遍历的顺序是:左子树 -> 根节点 -> 右子树;
后序遍历的顺序是:左子树 -> 右子树 -> 根节点。
以下是三种遍历方式的代码实现(假设二叉树的节点是一个包含val、left和right属性的对象):
```python
# 前序遍历
def preorderTraversal(root):
if not root:
return []
res = []
res.append(root.val)
res += preorderTraversal(root.left)
res += preorderTraversal(root.right)
return res
# 中序遍历
def inorderTraversal(root):
if not root:
return []
res = []
res += inorderTraversal(root.left)
res.append(root.val)
res += inorderTraversal(root.right)
return res
# 后序遍历
def postorderTraversal(root):
if not root:
return []
res = []
res += postorderTraversal(root.left)
res += postorderTraversal(root.right)
res.append(root.val)
return res
```
以上代码均为递归实现,也可以使用栈来实现迭代遍历。
阅读全文