在上篇博文《Java学习笔记之创建二叉树》后,我们现在来求增加二叉树的节点数、二叉树的深度的函数,以下代码中黄色背景是增加实现的代码,由于注释较多,我用绿色字体将自己解读的注解区分。
老样子还是先注明这句话:【本文的代码请见原创http://blog.csdn.net/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