剑指offer-27 二叉树的镜像
微信搜索【程序员画工师】关注更多Java编程技术、数据结构与算法、面试题相关内容。
题目
请完成一个函数,输入一个二叉树,输出它的镜像。镜像二叉树示意如下图
思路
以上图为例,先交换根节点的两个子节点,交换之后发现10、6的子节点仍然保持不变,所以下一步交换这两个节点的左、右子节点,交换之后的树就是原始树的镜像。递归解法的思路:
•先序遍历原二叉树的每个节点,如果遍历到的结点有子结点,就交换它的两个子结点。
•递归遍历每个节点的子节点,同样,如果遍历到的子节点有子节点,就交换它的两个子节点。
•当交换完所有非叶子结点的左右子结点之后,就得到了树的镜像。
上代码
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public void Mirror(TreeNode root) {
if(root == null){
return;
}
if(root.left == null && root.right == null){
return;
}
TreeNode tempNode = new TreeNode(0);
tempNode = root.left;
root.left = root.right;
root.right = tempNode;
if(root.left != null){
Mirror(root.left);
}
if(root.right != null){
Mirror(root.right);
}
}
}
扩展
也可以用循环的方法来实现,思路如下
•借助队列/栈等数据结构
•让二叉树每层的节点依次进入队列/栈
•队列/栈中的节点出队/栈,然后交换该节点的左右子节点
有兴趣的读者可以自己尝试实现。
References
[1] 《剑指offer(第二版)》 何海涛著
程序员画工师公众号,获取更多详细Java、Python、C、前端、小程序、产品相关学习资料,欢迎交流