- 标签:树、深度优先搜索、广度优先搜索、二叉树
- 难度:简单
描述:给定一个二叉树的根节点 root
。
要求:将该二叉树进行左右翻转。
说明:
- 树中节点数目范围在
$[0, 100]$ 内。 -
$-100 \le Node.val \le 100$ 。
示例:
- 示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
- 示例 2:
输入:root = [2,1,3]
输出:[2,3,1]
根据我们的递推三步走策略,写出对应的递归代码。
-
写出递推公式:
- 递归遍历翻转左子树。
- 递归遍历翻转右子树。
- 交换当前根节点
root
的左右子树。
-
明确终止条件:当前节点
root
为None
。 -
翻译为递归代码:
-
定义递归函数:
invertTree(self, root)
表示输入参数为二叉树的根节点root
,返回结果为翻转后二叉树的根节点。 -
书写递归主体:
left = self.invertTree(root.left) right = self.invertTree(root.right) root.left = right root.right = left return root
-
明确递归终止条件:
if not root: return None
-
-
返回根节点
root
。
class Solution:
def invertTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
if not root:
return None
left = self.invertTree(root.left)
right = self.invertTree(root.right)
root.left = right
root.right = left
return root
-
时间复杂度:$O(n)$,其中
$n$ 是二叉树的节点数目。 -
空间复杂度:$O(n)$。递归函数需要用到栈空间,栈空间取决于递归深度,最坏情况下递归深度为
$n$ ,所以空间复杂度为$O(n)$ 。