数据结构-二叉树(复制二叉树的递归算法)

文章目录

思路

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;
    }
}
发布了197 篇原创文章 · 获赞 62 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/abcnull/article/details/104450717