构建有序二叉树有两种构建方法:1、按大小顺序构建 2、使用递归
构建二叉树和构建链表类似
先写一个二叉树的节点类TreeNode类:
public class TreeNode {
//定义左子节点
public TreeNode leftNode;
//定义右子节点
public TreeNode rightNode;
//定义节点的权也就是节点的值
public Integer val;
//构造器来创建二叉树的某个节点
public TreeNode(Integer val) {
this.val=val;
}
}
1.我们用第一种不用递归的方法构建有序二叉树, 写一个有序二叉树类BinaryTree类:
public class BinaryTree {
//定义一个根节点
TreeNode root;
//构建有序二叉树的添加元素的方法
public void insert(Integer val) {
//创建一个新节点
TreeNode newNode=new TreeNode(val);
//判断根节点是否为空,是空就通过值传递把新节点赋值给根节点,
if(root==null) {
root=newNode;
//直接退出方法
return;
}
//定义一个游标指针,方便读取到该添加到的位置
TreeNode temp=root;
//定义一个父游标指针,永远指在游标指针的父节点
TreeNode partent=null;
while(true) {
//先让父游标partent记录temp游标指针的位置
partent=temp;
//判断新节点的位置是放在左边还是右边
//根据新节点值的大小判断新节点应该放在左节点还是右节点
if(temp.val<newNode.val) {
//让temp指针向下指
temp=temp.rightNode;
//temp指针没有子节点时将新节点添加在其后面
if(temp==null) {
partent.rightNode=newNode;
//必须写return,否则会陷入死循环
return;
}
}else {
temp=temp.leftNode;
if(temp==null) {
partent.leftNode=newNode;
return;
}
}
}
}
}
2.用递归的方式构建有序二叉树,我们接着在BinaryTree类里写一个递归的方法:
public TreeNode insertDigui(Integer val,TreeNode node) {
//创建一个新节点,为其分配空间
TreeNode newNode=new TreeNode(val);
//判断根节点有没有元素,没有就通过值传递将新节点赋值给根节点,直接return退出不再继续执行
if(root==null) {
return root=newNode;
}
//根据新节点值的大小判断新节点应该放在左节点还是右节点
if(val>node.val) {
//如果是大于某祖宗节点的值
//如果节点的右节点是空把新节点添加进去,return根节点
if(node.rightNode==null) {
node.rightNode=newNode;
return root;
}
//通过递归的方式,二叉树继续向下遍历
return insertDigui(val, node.rightNode);
}else {
//如果是小于某祖宗节点的值
if(node.leftNode==null) {
node.leftNode=newNode;
return root;
}
//通过递归的方式,二叉树继续向下遍历
return insertDigui(val, node.leftNode);
}
}
我们写一个测试类test测试一下:
public class Test {
public static void main(String[] args) {
BinaryTree binaryTree=new BinaryTree();
//按顺序大小进行构建的有序二叉树
binaryTree.insert(1);
binaryTree.insert(2);
binaryTree.insert(3);
binaryTree.insert(4);
binaryTree.insert(5);
//递归构建的有序二叉树
binaryTree.insertDigui(6, binaryTree.root);
binaryTree.insertDigui(7, binaryTree.root);
binaryTree.insertDigui(8, binaryTree.root);
binaryTree.insertDigui(9, binaryTree.root);
//写的一个广度优先遍历进行的输出
binaryTree.Order();
}
}
结果如下图所示: