要求
给你一个二叉树的根节点 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
重点问题
解题思路:
递归的方式进行构建,我们可以看到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,和上面的构建过程是一样的,不断递归进行构建即可。