使用先序遍历复制 Java 中的二叉树

Question

我正在尝试使用预序遍历复制二叉树，但我卡住了。由于我没有将任何值放入新树中，因此它们显然没有正确复制...

public class Node{

int key;
String name;

Node leftChild;
Node rightChild;

Node(int key, String name){
    this.key = key;
    this.name = name;

}



public class BinaryTree{

public Node root;

public void copyTree(Node focusNode){

    if(focusNode != null){

        Node copyNode = new Node(focusNode.key, focusNode.name);

        //System.out.println(copyNode);

        copyTree(focusNode.leftChild);
        copyTree(focusNode.rightChild);
    }
}

}

Answer 1

这是一种解决方案。为了演示目的，我向 Node class 添加了一个 toString() 方法。

class Node {

    int key;
    String name;

    Node leftChild;
    Node rightChild;

    Node(int key, String name) {
        this.key = key;
        this.name = name;
    }

    public String toString() {
        return "[" + key + "," + name + "]";
    }
}

BinaryTree也稍作修改：

class BinaryTree {

    public Node root;

    public BinaryTree copyTree(Node focusNode) {
        BinaryTree bt = new BinaryTree();
        bt.root = preOrderCopy(focusNode);
        return bt;
    }

    public static void preOrderPrint(BinaryTree t) {
        preOrderPrint(t.root);
    }

    public static void preOrderPrint(Node n) {
        if (n == null) {
            // base case
            return;
        }
        System.out.println(n);
        preOrderPrint(n.leftChild);
        preOrderPrint(n.rightChild);

    }

    private Node preOrderCopy(Node focusNode) {
        if (focusNode == null) {
            // base case
            return null;
        }
        Node copy = new Node(focusNode.key, focusNode.name);
        copy.leftChild = preOrderCopy(focusNode.leftChild);
        copy.rightChild = preOrderCopy(focusNode.rightChild);
        return copy;
    }

}

为了测试代码，我根据 Wikipedia page for Tree Traversal 上显示的代码创建了一个 BinaryTree。这是此示例中使用的树的图片：

此示例的正确预序遍历是：F, B, A, D, C, E, G, I, H。您可以使用以下代码来测试此实现：

public class NodeTest {

    public static void main(String[] args) {
        BinaryTree bt = new BinaryTree();
        Node a = new Node(1, "A");
        Node b = new Node(2, "B");
        Node c = new Node(3, "C");
        Node d = new Node(4, "D");
        Node e = new Node(5, "E");
        Node f = new Node(6, "F");
        Node g = new Node(7, "G");
        Node h = new Node(8, "H");
        Node i = new Node(9, "I");
        f.leftChild = b;
        b.leftChild = a;
        b.rightChild = d;
        d.leftChild = c;
        d.rightChild = e;
        f.rightChild = g;
        g.rightChild = i;
        i.leftChild = h;
        bt.root = f;

        System.out.println("Print full tree:");
        BinaryTree.preOrderPrint(bt.copyTree(f));

        System.out.println("Only print f's left sub-tree:");
        BinaryTree.preOrderPrint(bt.copyTree(f.leftChild));

    }

}

运行上述代码产生以下输出：

Print full tree:
[6,F]
[2,B]
[1,A]
[4,D]
[3,C]
[5,E]
[7,G]
[9,I]
[8,H]
Only print f's left sub-tree:
[2,B]
[1,A]
[4,D]
[3,C]
[5,E]

Answer 2

从a树复制到b树。你可以像我一样使用两个静态方法。我的想法来自添加 Element 方法和删除 Element 方法。他们很相似。

public static Node copyRec(Node a, Node b)//copy from b to a
{
    if(b!=null)
    {
        a=new Node(b.data);
        a.leftChild=copyRec(a.leftChild,b.leftChild);
        a.rightChild=copyRec(a.rightChild,b.rightChild);
        return a;
    }
    return null;
}
public static void copy(BST a, BST b)
{
    a.root=copyRec(a.root,b.root);
}

使用先序遍历复制 Java 中的二叉树

Copying a Binary Tree in Java using a preorder traversal

java

copy

binary-tree

preorder