class Solution {
public TreeNode mirrorTree(TreeNode root) {
//特殊结点处理
if(root == null) return root;
//终止条件,即最后一个点的处理
if(root.left == null && root.right == null) return root;
//递推公式部分
//左子树镜像操作
TreeNode left = mirrorTree(root.left);
//右子树镜像操作
TreeNode right = mirrorTree(root.right);
//左右子树交换
root.left = right;
root.right = left;
//返回交换完的子树
return root;
}
}
此代码很笨拙,套用的递归法解题步骤,还可以优化很多。
用递归解题的步骤有两个:
1.写递归公式
2.找终止条件
本题中:
递推公式:
左子树整体实现镜像操作,右子树整体实现镜像操作,左右子树交换位置,将根节点返回。
终止条件:
一般考虑最后一个模块即最后一个结点(叶子结点)的处理。当一个结点左子树和右子树都为null,说明是最后一个结点。
对于最后一个结点,直接return该结点即可。
注意:除了步骤中的两步,别忘了几乎每个题都要考虑的空值情况。结点为空值时,返回即可。
优化代码就是大家看到的那些大佬写的代码啦,理解了最笨的方法,优化之后的方法也自然就会看了。