二叉树遍历完整代码
时间: 2023-07-03 11:21:35 浏览: 52
二叉树的遍历有三种方式:前序遍历、中序遍历和后序遍历。下面是它们的完整代码实现。
```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()
res.append(node.val)
if node.right:
stack.append(node.right)
if node.left:
stack.append(node.left)
return res
# 中序遍历
def inorderTraversal(root):
if not root:
return []
res = []
stack = []
node = root
while stack or node:
if node:
stack.append(node)
node = node.left
else:
node = stack.pop()
res.append(node.val)
node = node.right
return res
# 后序遍历
def postorderTraversal(root):
if not root:
return []
res = []
stack = [root]
while stack:
node = stack.pop()
res.append(node.val)
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
return res[::-1]
```
这里采用了迭代的方式实现遍历,使用了栈来保存遍历的结点。对于前序遍历和后序遍历,我们需要在遍历时注意入栈的顺序,对于中序遍历,需要注意出栈的时机。