题目:请完成一个函数,输入一个二叉树,该函数输出它的镜像。
思路:该题目处于剑指offer第四章—画图让抽象问题形象化,我们可以先采用画图的方式求得其规律。
求一棵树镜像的过程:先前序遍历该树的每个结点,如果遍历到的结点有子结点,就交换它的两个子节点,当交换完所有非叶子结点的左右子节点之后,就得到了树的镜像。
注意:特殊输入(二叉树的根结点为null)
import java.util.*;
public class wr19Mirror {
// 递归
public void Mirror(TreeNode root){
if(root==null){
return;
}
if(root.left==null && root.right==null){
return ;
}
TreeNode temp=root.left;
root.left=root.right;
root.right=temp;
if(root.left!=null){
Mirror(root.left);
}
if(root.right!=null){
Mirror(root.right);
}
}
// 基于栈的非递归
public void stackMirror(TreeNode root){
if(root==null){
return ;
}
Stack<TreeNode> stack=new Stack<>();
stack.push(root);
while(!stack.isEmpty()){
TreeNode current=stack.pop();
TreeNode temp=current.left;
current.left=current.right;
current.right=temp;
if(current.left!=null){
stack.push(current.left);
}
if(current.right!=null){
stack.push(current.right);
}
}
}
}