用Python实现二叉树的遍历
时间: 2023-04-05 11:03:46 浏览: 69
可以回答这个问题。二叉树的遍历有三种方式:前序遍历、中序遍历和后序遍历。以下是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
```
这段代码使用了栈来实现前序遍历,先将根节点入栈,然后弹出栈顶元素,将其值加入结果列表中,再将右子节点和左子节点依次入栈。重复这个过程直到栈为空。
相关问题
python链表实现二叉树遍历
首先,我们需要创建一个节点类来表示二叉树中的节点:
```python
class Node:
def __init__(self, val):
self.val = val
self.left = None
self.right = None
```
然后,我们可以通过链表的方式来构建二叉树。例如,我们可以使用以下方式创建一个二叉树:
```python
root = Node(1)
root.left = Node(2)
root.right = Node(3)
root.left.left = Node(4)
root.left.right = Node(5)
root.right.left = Node(6)
root.right.right = Node(7)
```
接下来,我们可以使用递归的方式来实现二叉树的遍历。例如,我们可以使用以下方式实现前序遍历:
```python
def pre_order_traversal(root):
if root:
print(root.val)
pre_order_traversal(root.left)
pre_order_traversal(root.right)
```
类似地,我们可以使用以下方式实现中序遍历和后序遍历:
```python
def in_order_traversal(root):
if root:
in_order_traversal(root.left)
print(root.val)
in_order_traversal(root.right)
def post_order_traversal(root):
if root:
post_order_traversal(root.left)
post_order_traversal(root.right)
print(root.val)
```
这些遍历函数可以通过传入根节点来遍历整个二叉树。
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实现二叉树遍历的示例代码。