数据结构之二叉树篇卷一 -- 建立二叉树（With Java)

一、定义二叉树节点类

package tree;

public class Node<E> {
    public E data;
    public Node<E> lnode;
    public Node<E> rnode;

    public Node(){}
    public Node(E data) {
        this.data = data;
    }
}

　　通过泛型(generics)定义了一个公有的节点类，包含一个数据域 data，以及两个引用域 lnode 和 rnode。构造函数提供有参和无参默认两种。当然，因为二叉树节点这个类的适用范围是在关于二叉树的操作中，所以更加完美的做法是将其作为私有内部类定义。

 

二、定义二叉树主类

public class BiTree <E> {
    
    /** root node of the binary tree    */
    private Node<E> root;
    /** the gross of total leaves    */
    private int leaveGross;
    private int depth;
    private String name;
    
    public BiTree(String name) {
        this.name = name;
        root = new Node<>();
    }

    public Node<E> getRoot() { return root; }
    
    public int getLeaveGross() { return leaveGross; }

    public String getName() { return name; }

    public int getDepth() { return depth(root); }

    public int depth(Node<E> root){
        if(root != null) {
            return Math.max(depth(root.lnode), depth(root.rnode)) + 1;
        } else
            return 0;
    }
}

　　这里最主要的是根节点信息 root, 它是二叉树的构建、遍历、求深度、求叶子总数的根本。其他属性，可以有选择的定义。这里我递归写了一个求二叉树深度的方法。

 

 三、构造二叉树

　　在主类<code>BiTree</code>中定义建立二叉树的方法 -- generate()。

    /**
     * Construct a Binary Tree with a given sequence of data as form of array.
     * The <code>data</code> should be given as a complete binary tree.
     * @param data
     * @author SheepCore@MarshallLee
     */
    public void generate(E[] data){
        if(data.length == 0)
            return;
        int numData = data.length;
        Node<E>[] nodes = new Node[numData];

        for (int i = 0; i < numData; i++) {
            nodes[i] = new Node(data[i]);
        }

        for (int i = 0; i < numData / 2; i++) {
            nodes[i].lnode = nodes[2 * i + 1];
            nodes[i].rnode = nodes[2 * i + 2] ;
        }
        this.root = nodes[0];
        depth = getDepth();
        leaveGross = numData;
    }