思路
Java 对象满足深拷贝,直接赋值不就行了,为什么要复制呢?
确实,Java 的对象赋值满足深拷贝,类似 C 中的地址赋值,那这样直接 a=b 不就行了,干嘛还要写复制的一个算法呢?因为我们的这个复制希望 a 的操作不会影响 b 这颗二叉树,但是 a 和 b 这两颗二叉树中保存到的数据值都是一样的,这就是我们复制的目的
如何复制二叉树呢?
其实就是写一遍遍历而已
注意
我们传参传被拷贝的结点 r 和一个拷贝到的结点 n,如果 r 结点本身就是 null,那如果我们传的 n 结点不为 null,在我们不写else n=null;
时返回的 n 就不是空了,但我们期望它是空,这是一个小细节点,需要注意下,虽然不懈也不会影响代码逻辑,但是加上会显得代码更加严谨
Java 实现
// 结点
class Node {
int data;
Node left = null;
Node right = null;
}
// 二叉树
public class BinaryTree {
// 根结点
private Node root;
// 输入的数组
private int[] arr_in;
// 输出的数组
private int[] arr_out;
// 记录数组下标
private static int index;
// 初始化
public BinaryTree(int[] arr) {
root = new Node();
this.arr_in = arr;
arr_out = new int[arr.length];
index = 0;
}
// 先序复制二叉树(递归)根→左→右
public Node preorderCopyRecursion(Node r, Node n) {
if (r) {
n = new Node();
n = r.data;
preorderCopyRecursion(r.left, n.left);
preorderCopyRecursion(r.right, n.right);
}
else
n = null;
return n;
}
// 中序复制二叉树(递归)左→根→右
public Node inorderCopyRecursion(Node r, Node n) {
if (r) {
inorderCopyRecursion(r.left, n.left);
n = new Node();
n = r.data;
inorderCopyRecursion(r.right, n.right);
}
else
n = null;
return n;
}
// 后序复制二叉树(递归)左→右→根
public Node postorderCopyRecursion(Node r, Node n) {
if (r) {
postorderCopyRecursion(r.left, n.left);
postorderCopyRecursion(r.right, n.right);
n = new Node();
n = r.data;
}
else
n = null;
return n;
}
}