python反转二叉树

在数据结构和算法的学习中，二叉树是一种非常基础且重要的数据结构。反转二叉树是一个常见的操作，它指的是将二叉树的左右子树互换，而不改变节点的值。这个操作在很多情况下都非常有用，比如在某些特定的搜索算法优化中。本教程将详细介绍如何在Python中实现二叉树的反转。

一、定义二叉树节点

首先，我们需要定义一个二叉树的节点。每个节点包含三个部分：值（val）、左子节点（left）和右子节点（right）

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

二、反转二叉树的递归实现

递归是实现二叉树反转最直观的方法。基本思路是，首先递归地反转左子树和右子树，然后交换当前节点的左右子节点

def invertTree(root):  
    if root is None:  
        return None  
      
    # 递归反转左右子树  
    root.left, root.right = invertTree(root.right), invertTree(root.left)  
      
    # 返回反转后的根节点  
    return root

这个递归函数首先检查根节点是否为空，如果为空，则直接返回。然后，它递归地反转左子树和右子树，并交换当前节点的左右子节点。最后，返回反转后的根节点

三、反转二叉树的迭代实现

除了递归，我们还可以使用迭代的方式来实现二叉树的反转。迭代通常需要使用栈或队列等数据结构来辅助。

这里，我们使用栈来迭代反转二叉树

def invertTreeIterative(root):  
    if root is None:  
        return None  
      
    stack = [root]  # 使用栈来存储需要处理的节点  
      
    while stack:  
        node = stack.pop()  # 弹出栈顶元素  
        # 交换左右子节点  
        node.left, node.right = node.right, node.left  
          
        # 如果左子节点存在，则将其加入栈中  
        if node.left:  
            stack.append(node.left)  
        # 如果右子节点存在，则将其加入栈中  
        if node.right:  
            stack.append(node.right)  
      
    return root

四、测试代码

最后，我们需要编写一些测试代码来验证我们的反转函数是否正确

# 创建一个简单的二叉树  
#     4  
#    / \  
#   2   7  
#  / \ / \  
# 1  3 6  9  
  
root = TreeNode(4)  
root.left = TreeNode(2)  
root.right = TreeNode(7)  
root.left.left = TreeNode(1)  
root.left.right = TreeNode(3)  
root.right.left = TreeNode(6)  
root.right.right = TreeNode(9)  
  
# 使用递归方法反转二叉树  
inverted_root_recursive = invertTree(root)  
  
# 或者使用迭代方法  
# inverted_root_iterative = invertTreeIterative(root)  
  
# 打印反转后的二叉树（这里仅打印节点值以验证）  
def print_tree(root, level=0):  
    if root:  
        print_tree(root.right, level + 1)  
        print('  ' * level + '->', root.val)  
        print_tree(root.left, level + 1)  
  
print_tree(inverted_root_recursive)  # 或者打印 inverted_root_iterative

以上就是在Python中实现二叉树反转的详细教程，包括递归和迭代两种方法。希望这能帮助你更好地理解和实现二叉树的反转