python实现二叉树的遍历
时间: 2023-12-04 18:59:57 浏览: 85
python二叉树遍历的实现方法
二叉树是一种数据结构,其中每个节点最多有两个子节点。二叉树的遍历是指按照一定的顺序访问二叉树中的所有节点。常见的遍历方式有前序遍历、中序遍历和后序遍历。
以下是Python实现二叉树遍历的代码示例:
```python
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
class Solution:
def preorderTraversal(self, 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
def inorderTraversal(self, root: TreeNode) -> List[int]:
"""
中序遍历:左子树 -> 根节点 -> 右子树
"""
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(self, root: TreeNode) -> List[int]:
"""
后序遍历:左子树 -> 右子树 -> 根节点
"""
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]
```
上述代码中,我们定义了一个`TreeNode`类来表示二叉树节点,包括节点的值`val`、左子节点`left`和右子节点`right`。然后我们定义了一个`Solution`类,其中包含了三个函数,分别实现了前序遍历、中序遍历和后序遍历。
对于前序遍历,我们使用一个栈来存储节点,初始时将根节点加入栈中,然后循环执行以下操作:弹出栈顶节点并将其值加入结果列表,如果该节点的右子节点不为空,则将其右子节点加入栈中,如果该节点的左子节点不为空,则将其左子节点加入栈中。最后返回结果列表。
对于中序遍历,我们同样使用一个栈来存储节点,初始时将根节点加入栈中,然后循环执行以下操作:将当前节点的左子节点全部加入栈中,然后弹出栈顶节点并将其值加入结果列表,最后将当前节点置为其右子节点。如果当前节点为空且栈不为空,则弹出栈顶节点并继续执行上述操作。最后返回结果列表。
对于后序遍历,我们可以在前序遍历的基础上做一些修改,具体来说,先按照前序遍历的方式访问节点,并将节点加入结果列表中,然后将结果列表翻转即可得到后序遍历的结果。
以上就是Python实现二叉树遍历的示例代码。
阅读全文