二叉树的遍历 python

二叉树的遍历主要有三种方式：前序遍历、中序遍历和后序遍历。下面是使用 Python 实现这三种遍历的代码示例：

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

# 前序遍历
def preorderTraversal(root):
    if root is None:
        return []
    stack = [root]
    res = []
    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(root):
    if root is None:
        return []
    stack = []
    res = []
    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(root):
    if root is None:
        return []
    stack = [root]
    res = []
    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]

你可以通过创建二叉树节点的实例，并将其传入对应的函数来进行遍历。例如：

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

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

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

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

希望对你有所帮助！如果还有其他问题，请随时提问。