方式一:递归
我们下文以前序遍历为例,通过动画来看一下翻转的过程:
我们来看一下递归三部曲:
1、确定递归函数的参数和返回值
-
参数就是要传入节点的指针,不需要其他参数了,通常此时定下来主要参数,如果在写递归的逻辑中发现还需要其他参数的时候,随时补充。
-
返回值的话其实也不需要,但是题目中给出的要返回root节点的指针,可以直接使用题目定义好的函数,所以就函数的返回类型为TreeNode。
public TreeNode invertTree(TreeNode root) {}
2、确定终止条件
- 当前节点为空的时候,就返回
if (root == NULL) return root;
3、确定单层递归的逻辑
- 因为是先前序遍历,所以先进行交换左右孩子节点,然后反转左子树,反转右子树。
swap(root->left, root->right);
invertTree(root->left);
invertTree(root->right);
也就是交换当前节点的左右节点,再递归的交换当前节点的左节点,递归的交换当前节点的右节点。
//DFS递归
class Solution {
/**
* 前后序遍历都可以
* 中序不行,因为先左孩子交换孩子,再根交换孩子(做完后,右孩子已经变成了原来的左孩子),再右孩子交换孩子(此时其实是对原来的左孩子做交换)
*/
public TreeNode invertTree(TreeNode root) {
//递归函数的终止条件,节点为空时返回
if (root == null) {
return null;
}
//将当前节点的左右子树交换
swapChildren(root);
//递归交换当前节点的 左子树
invertTree(root.left);
//递归交换当前节点的 右子树
invertTree(root.right);
//函数返回时就表示当前这个节点,以及它的左右子树都已经交换完了
return root;
}
private void swapChildren(TreeNode root) {
//下面三句是将当前节点的左右子树交换
TreeNode tmp = root.left;
root.left = root.right;
root.right = tmp;
}
}
方式二:迭代
递归实现也就是深度优先遍历的方式,那么对应的就是广度优先遍历。 广度优先遍历需要额外的数据结构–队列,来存放临时遍历到的元素。 深度优先遍历的特点是一竿子插到底,不行了再退回来继续;而广度优先遍历的特点是层层扫荡。 所以,我们需要先将根节点放入到队列中,然后不断的迭代队列中的元素。对当前元素调换其左右子树的位置,然后:
- 判断其左子树是否为空,不为空就放入队列中
- 判断其右子树是否为空,不为空就放入队列中
动态图如下:
深度优先遍历和广度优先遍历,从动画图中看起来很类似,这是因为演示的树层数只有三层。
时间复杂度:同样每个节点都需要入队列/出队列一次,所以是 O(n)空间复杂度:最坏的情况下会包含所有的叶子节点,完全二叉树叶子节点是 n/2
个,所以时间复杂度是 0(n)代码实现如下:
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root==null) {
return null;
}
//将二叉树中的节点逐层放入队列中,再迭代处理队列中的元素
LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
queue.add(root);
while(!queue.isEmpty()) {
//每次都从队列中拿一个节点,并交换这个节点的左右子树
TreeNode tmp = queue.poll();
TreeNode left = tmp.left;
tmp.left = tmp.right;
tmp.right = left;
//如果当前节点的左子树不为空,则放入队列等待后续处理
if(tmp.left!=null) {
queue.add(tmp.left);
}
//如果当前节点的右子树不为空,则放入队列等待后续处理
if(tmp.right!=null) {
queue.add(tmp.right);
}
}
//返回处理完的根节点
return root;
}
}