温馨提示×

Python TreeNode如何进行树的遍历

小樊
104
2024-08-17 16:20:35
栏目: 编程语言

在Python中,可以通过递归的方式来实现树的遍历。下面是一个示例代码,演示了如何实现 TreeNode 类和对其进行前序、中序和后序遍历。

class TreeNode:
    def __init__(self, value):
        self.value = value
        self.left = None
        self.right = None

def preorder_traversal(root):
    if root:
        print(root.value, end=' ')
        preorder_traversal(root.left)
        preorder_traversal(root.right)

def inorder_traversal(root):
    if root:
        inorder_traversal(root.left)
        print(root.value, end=' ')
        inorder_traversal(root.right)

def postorder_traversal(root):
    if root:
        postorder_traversal(root.left)
        postorder_traversal(root.right)
        print(root.value, end=' ')

# 构建一个示例树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

print("Preorder traversal:")
preorder_traversal(root)
print("\nInorder traversal:")
inorder_traversal(root)
print("\nPostorder traversal:")
postorder_traversal(root)

以上代码演示了如何定义一个简单的 TreeNode 类,以及如何进行前序、中序和后序遍历。你可以根据自己的需要对以上代码进行修改和扩展。

0