python二叉树遍历算法怎么实现-开发者知识库平台

python二叉树遍历算法怎么实现

python

1174

2024/2/26 11:33:33

栏目: 编程语言

在Python中，可以使用递归实现二叉树的三种遍历算法：前序遍历、中序遍历和后序遍历。

下面是一个简单的二叉树节点类的定义：

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

接下来，我们可以定义三个函数分别实现前序遍历、中序遍历和后序遍历算法：

前序遍历（Preorder traversal）：首先访问根节点，然后递归遍历左子树，最后递归遍历右子树。

def preorderTraversal(root):
    result = []
    if root:
        result.append(root.val)
        result += preorderTraversal(root.left)
        result += preorderTraversal(root.right)
    return result

中序遍历（Inorder traversal）：先递归遍历左子树，然后访问根节点，最后递归遍历右子树。

def inorderTraversal(root):
    result = []
    if root:
        result += inorderTraversal(root.left)
        result.append(root.val)
        result += inorderTraversal(root.right)
    return result

后序遍历（Postorder traversal）：先递归遍历左子树，然后递归遍历右子树，最后访问根节点。

def postorderTraversal(root):
    result = []
    if root:
        result += postorderTraversal(root.left)
        result += postorderTraversal(root.right)
        result.append(root.val)
    return result

接下来，我们可以创建一个二叉树，并使用上述函数进行遍历：

# 创建一个二叉树
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

# 前序遍历
print(preorderTraversal(root))  # Output: [1, 2, 4, 5, 3]

# 中序遍历
print(inorderTraversal(root))   # Output: [4, 2, 5, 1, 3]

# 后序遍历
print(postorderTraversal(root)) # Output: [4, 5, 2, 3, 1]

以上就是用Python实现二叉树的三种遍历算法的方法。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

python二叉树遍历算法怎么实现

最新知识库

相关标签