Java学习笔记之二叉树的节点数、深度

在上篇博文《Java学习笔记之创建二叉树》后，我们现在来求增加二叉树的节点数、二叉树的深度的函数，以下代码中黄色背景是增加实现的代码，由于注释较多，我用绿色字体将自己解读的注解区分。

老样子还是先注明这句话：【本文的代码请见原创http://blog.youkuaiyun.com/wuwenxiang91322/article/details/12231657】

我是结合他的代码一行行通过注释解读自己的理解，从中温习java的语法结构及数据结构、算法的思想。如有错误，望批评指正~

package tree;
import java.util.Stack;

/**
 * 二叉树的链式存储
 * @author WWX
 */
public class BinaryTree{
//二叉树通常用树结点结构存储，有时也包含指向唯一父节点的指针
private TreeNode root = null;

//BinTree()该方法与类名字相同，所以是构造方法，被默认强制为void
public  BinaryTree(){
root = new TreeNode(1,"A");
}

/*创建二叉树，树由结点构成
* <pre>
*           A
*     B          C
*  D     E            F
*  </pre>
* @param root
* @author WWX
*/
public void createBinTree(TreeNode root){
TreeNode newNodeB = new TreeNode(2,"B");
TreeNode newNodeC = new TreeNode(3,"C");
TreeNode newNodeD = new TreeNode(4,"D");
TreeNode newNodeE = new TreeNode(5,"E");
TreeNode newNodeF = new TreeNode(6,"F");
root.leftChild = newNodeB;
root.rightChild = newNodeC;
root.leftChild.leftChild = newNodeD;
root.leftChild.rightChild = newNodeE;
root.rightChild.rightChild = newNodeF;

}

//节点个数

//二叉树的递归思想，先定义根节点root
public int size(){
return size(root);
}
//二叉树递归思想：定义根节点root后，再用subtree实现递归
private int size(TreeNode subtree){
if (subtree == null)
return 0;
else
{
return 1+size(subtree.leftChild)+size(subtree.rightChild);
}
}

//树的高度
//二叉树递归思想，先定义root
public int height(){
return height(root);

}
//二叉树递归思想，定义root后，再用subtree实现递归
private int height(TreeNode subtree){
if (subtree == null)
return 0;
else {
int i = height(subtree.leftChild);
int j = height(subtree.rightChild);
return (i<j)?(j+1):(i+1);//三目运算符知识，如果(i<j)是真的，则返回j+1,否则返回i+1
}

}

/**
* 二叉树的节点数据结构
* @author WWX
*/
private class TreeNode{
private int key = 0; //key 为层序编码
private String data = null; //data 为数据域
private boolean isVisted = false;

/*     树的每一个节点的数据结构都是TreeNode类型，
createBinTree里定义的root为TreeNode类型，所以左右孩子也为TreeNode类型，
加上二叉树的递归思想，所以所有节点都是TreeNode类型
        */ 
private TreeNode leftChild = null;
private TreeNode rightChild = null;



/*TreeNode(int key,String data) 方法名与类名相同，所以为构造方法
* 
 为什么要用到this？
 因为在TreeNode()函数中需要该函数所属类的当前对象

//这里符合必须使用this的情况：
* 构造方法将外部传入的参数赋值给类的成员变量
* 构造方法的形式参数名称与类的成员变量名一致
*/

//构造方法的形式参数名称与类的成员变量名一致
public TreeNode(int key,String data){

this.key = key;//构造方法将外部传入的参数赋值给类的成员变量
this.data = null;
this.isVisted = false;
this.leftChild = null;
this.rightChild = null;

}
}


//测试
public static void main(String[] args) {
        BinaryTree bt = new BinaryTree();
        //TreeNode root = new TreeNode(0, "R"); 
        
      /*这里方法里需要bt.root是由于root变量在BinaryTree类里定义的，

       *根据作用域定义，root只在BinaryTree类里起作用，其它地方调用root，都需要  bt.root
       * 比如：如果写成bt.createBinTree(root),就变成root就不属于BinarTree类了
      */
        bt.createBinTree(bt.root);
        System.out.println("the size of the tree is " + bt.size());
       System.out.println("the height of the tree is " + bt.height());
        
    }
}
 

打印的结果如下：

the size of the tree is 6
the height of the tree is 3