leetcode_156 上下翻转二叉树

376 阅读1分钟

要求

给你一个二叉树的根节点 root ,请你将此二叉树上下翻转,并返回新的根节点。

你可以按下面的步骤翻转一棵二叉树:

  • 原来的左子节点变成新的根节点
  • 原来的根节点变成新的右子节点
  • 原来的右子节点变成新的左子节点

上面的步骤逐层进行。题目数据保证每个右节点都有一个同级节点(即共享同一父节点的左节点)且不存在子节点。

示例 1:

输入:root = [1,2,3,4,5]
输出:[4,5,2,null,null,3,1]

示例 2:

输入:root = []
输出:[]

示例 3:

输入:root = [1]
输出:[1]

提示:

  • 树中节点数目在范围 [0, 10] 内
  • 1 <= Node.val <= 10
  • 树中的每个右节点都有一个同级节点(即共享同一父节点的左节点)
  • 树中的每个右节点都没有子节点

核心代码

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right
        
class Solution:
    def upsideDownBinaryTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root or (not root.left and not root.right):
            return root
        newroot = self.upsideDownBinaryTree(root.left)

        root.left.left = root.right
        root.left.right = root
        root.left = None
        root.right = None
        return newroot

image.png

重点问题

解题思路:

递归的方式进行构建,我们可以看到root = 1的时候,递归到root=2,递归到root=4,当root=4的时候,进入递归出口,我们返回的root=4,即newroot的值是4,此时root=2,我们看root.left(4).left=root.right(5),root.left(4)right=root(2),也就是4作为根节点然后左子树是5,右子树是2,和上面的构建过程是一样的,不断递归进行构建即可。