欢迎访问宙启技术站
智能推送

Python中如何实现树的后序遍历算法

发布时间:2023-12-27 20:15:54

在Python中,可以使用递归或迭代的方式实现树的后序遍历算法。

1. 递归方法:

递归方法是通过递归的方式实现树的后序遍历。下面是一个示例代码:

# 定义树节点类
class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

# 后序遍历函数
def postorderTraversal(root):
    if root is None:
        return []
    result = []
    result.extend(postorderTraversal(root.left))
    result.extend(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)
root.right.left = TreeNode(6)
root.right.right = TreeNode(7)

# 输出后序遍历结果
print(postorderTraversal(root))   # 输出: [4, 5, 2, 6, 7, 3, 1]

2. 迭代方法:

迭代方法是通过使用栈来实现树的后序遍历。下面是一个示例代码:

# 定义树节点类
class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

# 后序遍历函数
def postorderTraversal(root):
    if root is None:
        return []
    result = []
    stack = []
    curr = root
    while curr or stack:
        while curr:
            stack.append(curr)
            curr = curr.left if curr.left else curr.right
        node = stack.pop()
        result.append(node.val)
        if stack and stack[-1].left == node:
            curr = stack[-1].right
        else:
            curr = None
    return result

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

# 输出后序遍历结果
print(postorderTraversal(root))   # 输出: [4, 5, 2, 6, 7, 3, 1]

以上是使用递归和迭代的两种方法实现树的后序遍历算法。其中递归方法是比较常见的实现方式,而迭代方法则使用了栈的辅助结构来进行前序遍历。无论是哪种方法,后序遍历的顺序都是先遍历左子树,再遍历右子树,最后访问根节点。