1.题目
剑指 Offer 27. 二叉树的镜像
请完成一个函数,输入一个二叉树,该函数输出它的镜像。
例如输入:
4
/ \
2 7
/ \ / \
1 3 6 9
镜像输出:
4
/ \
7 2
/ \ / \
9 6 3 1
示例 1:
输入:root = [4,2,7,1,3,6,9]
输出:[4,7,2,9,6,3,1]
限制:
0 <= 节点个数 <= 1000
2.自我思路与实现
1.递归
很显然这是个递归的过程
树的根不变,左右子树交换
左子树的两个子树交换
右子树的两个子树交换
…
递归的终止条件是一个节点的两个子节点均为空
注意:当只有左子节点为空时,继续递归右子节点,右子节点为空同理
时间N: 递归遍历到了所有节点
空间N:递归的深度最大为N
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode invertTree(TreeNode root) {
if(root == null)
return root;
return invert(root);
}
public TreeNode invert(TreeNode p)
{
if(p.left == null && p.right == null)
return p;
if(p.left != null)
p.left = invert(p.left);
if(p.right != null)
p.right = invert(p.right);
TreeNode temp = p.left;
p.left = p.right;
p.right = temp;
return p;
}
}
3.总结思路与实现
一种更简洁的递归
class Solution {
public TreeNode invertTree(TreeNode root) {
if (root == null) {
return null;
}
TreeNode left = invertTree(root.left);
TreeNode right = invertTree(root.right);
root.left = right;
root.right = left;
return root;
}
}
改进有以下几点
- 直接判断节点是否为空,这样省去了判断节点的左右子树为空的三种情况
- 将反转后的左右子树直接定义为新的左右子树,再链接到根上,省去了引入第三个临时节点
- 直接递归本函数,省去建立新函数的过程
4.相关知识
- 递归的三个元素:递归函数的作用、递归的终止条件、递归关系式