剑指Offer 27.二叉树的镜像
利用栈或者队列实现:
将根节点入队,再出队,将该结点的左右子树入队,交换该结点的左右子树,再将队列中的结点出队,交换该结点的左右子树,重复此步骤,直到队列为空。返回根节点。
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if(root == null) return null;
Queue<TreeNode> tree = new LinkedList<>();
tree.add(root);
while(!tree.isEmpty()) {
TreeNode node = tree.remove();
if(node.left != null) tree.add(node.left);
if(node.right != null) tree.add(node.right);
TreeNode temp = node.left;
node.left = node.right;
node.right = temp;
}
return root;
}
}
利用递归实现:
递归解析:
终止条件: 当节点 rootroot 为空时(即越过叶节点),则返回 null ;
递推工作:
初始化节点 tmp ,用于暂存 root的左子节点;
开启递归 右子节点 mirrorTree(root.right),并将返回值作为 rootroot 的 左子节点 。
开启递归 左子节点 mirrorTree(tmp) ,并将返回值作为 rootroot 的 右子节点 。
返回值: 返回当前节点 rootroot ;
Q: 为何需要暂存 rootroot 的左子节点?
A: 在递归右子节点 “root.left = mirrorTree(root.right);root.left=mirrorTree(root.right);” 执行完毕后, root.leftroot.left 的值已经发生改变,此时递归左子节点 mirrorTree(root.left)mirrorTree(root.left) 则会出问题。
class Solution {
public TreeNode mirrorTree(TreeNode root) {
if(root == null) return null;
TreeNode temp = root.left;
root.left = mirrorTree(root.right);
root.right = mirrorTree(temp);
return root;
}
}