题目描述:
给你一棵二叉树的根节点 root ,翻转这棵二叉树,并返回其根节点。
示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
示例 2:
输入:root = [2,1,3]
输出:[2,3,1]
示例 3:
输入:root = []
输出:[]
思路:
解法1:递归
1、终止条件: 当节点 rootrootroot 为空时(即越过叶节点),则返回 nullnullnull 。
2、递推工作:
3、初始化节点 tmptmptmp ,用于暂存 rootrootroot 的左子节点。
开启递归 右子节点 invertTree(root.right)invertTree(root.right)invertTree(root.right) ,并将返回值作为 rootrootroot 的 左子节点 。
开启递归 左子节点 invertTree(tmp)invertTree(tmp)invertTree(tmp) ,并将返回值作为 rootrootroot 的 右子节点 。
4、返回值: 返回当前节点 rootrootroot 。
注意:为何需要暂存 rootrootroot 的左子节点?
在递归右子节点 “root.left=invertTree(root.right);root.left = invertTree(root.right);
root.left=invertTree(root.right);” 执行完毕后, root.leftroot.leftroot.left 的值已经发生改变,此时递归左子节点 invertTree(root.left)invertTree(root.left)invertTree(root.left) 则会出问题。
python:
# 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
tmp=root.left
root.left=self.invertTree(root.right)
root.right=self.invertTree(tmp)
return root
复杂度分析:
时间复杂度 O(N): 其中 N 为二叉树的节点数量,建立二叉树镜像需要遍历树的所有节点,占用 O(N)时间。
空间复杂度 O(N): 最差情况下(当二叉树退化为链表),递归时系统需使用 O(N) 大小的栈空间。
解法2:栈
利用栈遍历树的所有节点 node,并交换每个 node的左 / 右子节点。
步骤:
1、 当 root为空时,直接返回 null。
2、初始化:用栈,并加入根节点 root 。
3、循环交换: 当栈 stack 为空时跳出。
出栈: 记为 node。
添加子节点: 将 node左和右子节点入栈。
交换: 交换 node 的左 / 右子节点。
4、返回值: 返回根节点 root。
python:
# 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
stack=[root]
while stack:
node=stack.pop()
if node.left:
stack.append(node.left)
if node.right:
stack.append(node.right)
node.left,node.right=node.right,node.left
return root