题目介绍
力扣226题:https://leetcode-cn.com/problems/invert-binary-tree/
翻转一棵二叉树。
分析
这是一道很经典的二叉树问题。显然,我们可以遍历这棵树,分别翻转左右子树,一层层递归调用,就可以翻转整个二叉树了。
方法一:先序遍历
容易想到,我们可以先考察根节点,把左右子树调换,然后再分别遍历左右子树、依次翻转每一部分就可以了。这对应的遍历方式,就是先序遍历。
代码如下:
// 1. 先序遍历
public TreeNode invertTree1(TreeNode root){
// 处理基准场景
if (root == null) return null;
// 1. 先处理根节点,交换左右子节点
TreeNode temp = root.left;
root.left = root.right;
root.right = temp;
// 2. 递归处理左右子树
invertTree(root.left);
invertTree(root.right);
return root;
}
复杂度分析
- 时间复杂度:O(N),其中 N 为二叉树节点的数目。我们会遍历二叉树中的每一个节点,对每个节点而言,我们在常数时间内交换其两棵子树。
- 空间复杂度:O(logN)。使用的空间由递归栈的深度决定,它等于当前节点在二叉树中的高度。在平均情况下,二叉树的高度与节点个数为对数关系,即O(logN)。而在最坏情况下,树形成链状,空间复杂度为O(N)。
方法二:后序遍历
类似地,我们也可以用后序遍历的思路:先递归地处理左右子树,然后再将左右子树调换就可以了。
代码如下:
// 2. 后序遍历
public TreeNode invertTree(TreeNode root){
// 处理基准场景
if (root == null) return null;
// 1. 递归处理左右子树
TreeNode left = invertTree(root.left);
TreeNode right = invertTree(root.right);
// 2. 处理根节点,交换左右子节点
root.left = right;
root.right = left;
return root;
}
复杂度分析略,与方法一完全相同。