写一段二叉树遍历的代码
时间: 2023-12-01 13:01:13 浏览: 75
二叉树,数列等基本代码
以下是二叉树遍历的代码,包括先序遍历、中序遍历和后序遍历:
```python
# 定义二叉树的节点
class TreeNode:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
# 先序遍历
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:
while node:
stack.append(node)
node = node.left
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]
```
其中,先序遍历使用了栈进行迭代,中序遍历使用了栈和指针进行迭代,后序遍历使用了栈进行迭代,并且最后将结果翻转。
阅读全文