Python, LintCode, 175. 翻转二叉树

本文介绍了一种翻转二叉树的递归算法实现,该算法通过交换二叉树节点的左右子节点来达到完全翻转整棵树的目的。文章详细展示了如何定义二叉树节点,并提供了具体的Python代码实现。
"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""

class Solution:
    """
    @param root: a TreeNode, the root of the binary tree
    @return: nothing
    """
    def invertBinaryTree(self, root):
        # write your code here
        if root is None:
            return None
        temproot = self.invertBinaryTree(root.left)
        root.left = self.invertBinaryTree(root.right)
        root.right = temproot
        return root

### Python 翻转二叉树实现方法 #### 使用递归方式翻转二叉树 定义 `TreeNode` 类用于表示二叉树中的节点。该类具有三个属性:`val` 表示节点值;`left` 和 `right` 分别指向左子树和右子树。 ```python class TreeNode: def __init__(self, val=0, left=None, right=None): self.val = val self.left = left self.right = right ``` 创建解决方案类 `Solution` 并定义成员函数 `invertTree` 来执行二叉树的反转操作。如果当前处理的是空节点,则直接返回此节点本身作为终止条件。对于非空节点,先交换其左右孩子的位置再分别对其左右子树调用 `invertTree` 函数完成整棵树的镜像变换[^2]。 ```python class Solution: def invertTree(self, root: TreeNode) -> TreeNode: if not root: return None # 交换左右子树 root.left, root.right = root.right, root.left # 对新的左右子树继续进行倒置 self.invertTree(root.left) self.invertTree(root.right) return root ``` #### 迭代法实现翻转二叉树 除了上述递归的方式外还可以采用栈来模拟递归来达到相同的效果。初始化时将根结点压入栈内,在循环过程中不断弹出最上面的一个元素并将其左右孩子的顺序颠倒过来之后再次把它们依次放入栈底直到遍历完整棵二叉树为止[^1]。 ```python from collections import deque def invert_tree_iteratively(root): stack = [root] while stack: node = stack.pop() if node is not None: # 将左右子节点加入到待访问列表中 stack.append(node.left) stack.append(node.right) # 互换左右子节点位置 node.left, node.right = node.right, node.left return root ```
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值