leetcode-翻转二叉树-CSDN博客

本文链接：https://blog.csdn.net/weixin_33631777/article/details/135966423

本文介绍了如何使用前序遍历和迭代法在Python中实现二叉树的翻转，通过交换每个节点的左右孩子来达到翻转的效果。给出了两种解决方案，包括递归和迭代版本。

摘要由CSDN通过智能技术生成

题解：

发现想要翻转它，其实把每个节点的左右孩子交换一下就可以了。

我们使用前序遍历，先进行交换左右孩子节点，然后翻转左子树，翻转右子树

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root:
            return None
        root.left, root.right = root.right, root.left
        self.invertTree(root.left)
        self.invertTree(root.right)
        return root

迭代法

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root:
            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