二叉查找树的定义:
二叉查找树或者是一颗空树,或者是一颗具有以下特性的非空二叉树:
1. 若左子树非空,则左子树上所有节点关键字值均小于根节点的关键字;
2. 若右子树非空,则右子树上所有节点关键字值均大于根节点的关键字;
3. 左、右子树本身也分别是一颗二叉查找树。
二叉查找树的实现,功能有:
1. 用一个数组去构建二叉查找树
2. 二叉查找树的中序遍历和层次遍历
3. 插入节点
4. 查找节点
5. 查找二叉树中的最大值和最小值
6. 得到节点的直接父节点
7. 得到节点的直接前驱和直接后继节点
8. 删除节点
树节点TreeNode的定义见:Java实现链式存储的二叉树
import java.lang.Integer;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Scanner; /*
* 二叉排序树(二叉查找树)的实现
*/
public class BinarySearchTree {
private TreeNode<Integer> root; //根节点 public BinarySearchTree(){
root = null;
} //用一个数组去构建二叉查找树
public TreeNode<Integer> buildBST(Integer[] array){
if(array.length == 0){
return null;
}else{
root = null; //初始化树为空树
for(int i=0; i<array.length; i++){ //依次将每个元素插入
root = insertNode(root,array[i]);
}
return root;
}
} //在二叉查找树中插入一个数据域为data的结点,新插入的结点一定是某个叶子节点
public TreeNode<Integer> insertNode(TreeNode<Integer> node, Integer data){
if(node == null){ //原树为空,新插入的记录为根节点
node = new TreeNode<Integer>(data,null,null);
}else{
if(node.getData() == data){ //树中存在相同关键字的结点,什么也不做 }else{
if(node.getData() > data){ //根节点>插入数据,插入到左子树中
node.setLchild(insertNode(node.getLchild(),data));
}else{ //根节点<插入数据,插入到右子树中
node.setRchild(insertNode(node.getRchild(),data));
}
}
}
return node;
} //二叉查找树的中序遍历,可以得到一个递增的有序数列
public void inOrder(TreeNode<Integer> node){
if(node != null){
inOrder(node.getLchild());
System.out.print(node.getData()+" ");
inOrder(node.getRchild());
}
}
//二叉查找树的层次遍历
public void levelOrder(TreeNode<Integer> root){
Queue<TreeNode<Integer>> nodeQueue = new LinkedList<TreeNode<Integer>>();
TreeNode<Integer> node = null;
nodeQueue.add(root); //将根节点入队
while(!nodeQueue.isEmpty()){ //队列不空循环
node = nodeQueue.peek();
System.out.print(node.getData()+" ");
nodeQueue.poll(); //队头元素出队
if(node.getLchild() != null){ //左子树不空,则左子树入队列
nodeQueue.add(node.getLchild());
}
if(node.getRchild() != null){ //右子树不空,则右子树入队列
nodeQueue.add(node.getRchild());
}
}
} //查找数据域为data的结点,若不存在,返回null
public TreeNode<Integer> searchNode(TreeNode<Integer> node, Integer data){
while(node != null && node.getData() != data){
if(node.getData() > data){
node = node.getLchild(); //根节点>数据,向左走
}else{
node = node.getRchild(); //根节点<数据,向右走
}
}
return node;
} //查找最大值:不断地寻找右子节点
public TreeNode<Integer> getMaxData(TreeNode<Integer> node){
if(node.getRchild() == null){
return node;
}else{
return getMaxData(node.getRchild());
}
} //查找最小值:不断地寻找左子节点
public TreeNode<Integer> getMinData(TreeNode<Integer> node){
if(node.getLchild() == null){
return node;
}else{
return getMinData(node.getLchild());
}
} //得到数据域为data的结点的直接父节点parentNode
public TreeNode<Integer> getParentNode(TreeNode<Integer> root, Integer data){
TreeNode<Integer> parentNode = root;
if(parentNode.getData() == data){ //根节点的父节点返回为null
return null;
}
while(parentNode != null){
//查找当前节点的父节点的左右子节点,若是相等,则返回该父节点
if((parentNode.getLchild() != null && parentNode.getLchild().getData() == data) ||
(parentNode.getRchild() != null && parentNode.getRchild().getData() == data)){
return parentNode;
}else{
if(parentNode.getData() > data){ //向左查找父节点
parentNode = parentNode.getLchild();
}else{
parentNode = parentNode.getRchild(); //向右查找父节点
}
}
}
return null;
} /**
* 得到结点node的直接前趋
* a.该节点左子树不为空:其前驱节点为其左子树的最大元素
* b.该节点左子树为空: 其前驱节点为其祖先节点(递归),且该祖先节点的右孩子也为其祖先节点
* (就是一直往其parent找,出现左拐后的那个祖先节点)
*/
public TreeNode<Integer> getPrecessor(TreeNode<Integer> root,TreeNode<Integer> node){
if(node == null){
return null;
}
//a.该节点左子树不为空:其前驱节点为其左子树的最大元素
if(node.getLchild() != null){
return getMaxData(node.getLchild());
}else{ //b.该节点左子树为空: 其前驱节点为其祖先节点(递归)
TreeNode<Integer> parentNode = getParentNode(root, node.getData());
while(parentNode != null && node == parentNode.getLchild()){
node = parentNode;
parentNode = getParentNode(root, parentNode.getData());
}
return parentNode;
}
} /**
* 得到结点node的直接后继(后继节点就是比要删除的节点的关键值要大的节点集合中的最小值)
* a.该节点右子树不为空,其后继节点为其右子树的最小元素
* b.该节点右子树为空,则其后继节点为其祖先节点(递归),且此祖先节点的左孩子也是该节点的祖先节点,
* 就是说一直往上找其祖先节点,直到出现右拐后的那个祖先节点:
*/
public TreeNode<Integer> getSuccessor(TreeNode<Integer> root,TreeNode<Integer> node){
if(node == null){
return null;
}
//a.该节点右子树不为空,其后继节点为其右子树的最小元素
if(node.getRchild() != null){
return getMinData(node.getRchild());
}else{ //b.该节点右子树为空,则其后继节点为其最高祖先节点(递归)
TreeNode<Integer> parentNode = getParentNode(root, node.getData());
while(parentNode != null && node == parentNode.getRchild()){
node = parentNode;
parentNode = getParentNode(root, parentNode.getData());
}
return parentNode;
}
} /**
* 删除数据域为data的结点
* 按三种情况处理:
* a.如果被删除结点z是叶子节点,则直接删除,不会破坏二叉查找树的性质
* b.如果节点z只有一颗左子树或右子树,则让z的子树成为z父节点的子树,代替z的位置
* c.若结点z有左、右两颗子树,则令z的直接后继(或直接前驱)替代z,
* 然后从二叉查找树中删去这个直接后继(或直接前驱),这样就转换为第一或第二种情况
* @param node 二叉查找树的根节点
* @param data 需要删除的结点的数据域
* @return
*/
public boolean deleteNode(TreeNode<Integer> node, Integer data){
if(node == null){ //树为空
throw new RuntimeException("树为空!");
}
TreeNode<Integer> delNode= searchNode(node, data); //搜索需要删除的结点
TreeNode<Integer> parent = null;
if(delNode == null){ //如果树中不存在要删除的关键字
throw new RuntimeException("树中不存在要删除的关键字!");
}else{
parent = getParentNode(node,data); //得到删除节点的直接父节点
//a.如果被删除结点z是叶子节点,则直接删除,不会破坏二叉查找树的性质
if(delNode.getLchild()==null && delNode.getRchild()==null){
if(delNode==parent.getLchild()){ //被删除节点为其父节点的左孩子
parent.setLchild(null);
}else{ //被删除节点为其父节点的右孩子
parent.setRchild(null);
}
return true;
}
//b1.如果节点z只有一颗左子树,则让z的子树成为z父节点的子树,代替z的位置
if(delNode.getLchild()!=null && delNode.getRchild()==null){
if(delNode==parent.getLchild()){ //被删除节点为其父节点的左孩子
parent.setLchild(delNode.getLchild());
}else{ //被删除节点为其父节点的右孩子
parent.setRchild(delNode.getLchild());
}
delNode.setLchild(null); //设置被删除结点的左孩子为null
return true;
}
//b2.如果节点z只有一颗右子树,则让z的子树成为z父节点的子树,代替z的位置
if(delNode.getLchild()==null && delNode.getRchild()!=null){
if(delNode==parent.getLchild()){ //被删除节点为其父节点的左孩子
parent.setLchild(delNode.getRchild());
}else{ //被删除节点为其父节点的右孩子
parent.setRchild(delNode.getRchild());
}
delNode.setRchild(null); //设置被删除结点的右孩子为null
return true;
}
//c.若结点z有左、右两颗子树,则删除该结点的后继结点,并用该后继结点取代该结点
if(delNode.getLchild()!=null && delNode.getRchild()!=null){
TreeNode<Integer> successorNode = getSuccessor(node,delNode); //得到被删除结点的后继节点
deleteNode(node,successorNode.getData()); //删除该结点的后继结点
delNode.setData(successorNode.getData()); //用该后继结点取代该结点
return true;
}
}
return false;
} public static void main(String args[]){
Scanner input = new Scanner(System.in);
Integer[] array = {8,3,10,1,6,14,4,7,13};
BinarySearchTree bst = new BinarySearchTree();
TreeNode<Integer> root = bst.buildBST(array);
System.out.print("层次遍历:");
bst.levelOrder(root);
System.out.print("\n"+"中序遍历:");
bst.inOrder(root);
System.out.println();
System.out.print("得到最大值:");
System.out.println(bst.getMaxData(root).getData());
System.out.print("得到最小值:");
System.out.println(bst.getMinData(root).getData());
System.out.print("向二叉查找树中插入一个节点,请输入需插入节点的数据域:");
int data = input.nextInt();
System.out.print("插入节点"+ data +"后,中序遍历的结果:");
root = bst.insertNode(root, data);
bst.inOrder(root);
System.out.println("\n"+"在二叉查找树中查找元素,"+"请输入需要查找的结点值:");
data = input.nextInt();
if(bst.searchNode(root, data) == null){
System.out.println("false");
}else{
System.out.println("true");
}
System.out.println("查找节点的直接父节点,"+"请输入需要查找的结点值:");
data = input.nextInt();
System.out.print("节点"+ data +"的父节点是:");
if(bst.getParentNode(root, data) == null){
System.out.println("null");
}else{
System.out.println(bst.getParentNode(root, data).getData());
}
System.out.println("删除结点,"+"请输入需要删除的结点值:");
data = input.nextInt();
if(bst.deleteNode(root, data)){
System.out.print("删除结点后的层次遍历:");
bst.levelOrder(root);
System.out.print("\n"+"删除结点后的中序遍历:");
bst.inOrder(root);
}
}
}
程序运行结果:
层次遍历:8 3 10 1 6 14 4 7 13
中序遍历:1 3 4 6 7 8 10 13 14
得到最大值:14
得到最小值:1
向二叉查找树中插入一个节点,请输入需插入节点的数据域:15
插入节点15后,中序遍历的结果:1 3 4 6 7 8 10 13 14 15
在二叉查找树中查找元素,请输入需要查找的结点值:
4
true
查找节点的直接父节点,请输入需要查找的结点值:
10
节点10的父节点是:8
删除结点,请输入需要删除的结点值:
4
删除结点后的层次遍历:8 3 10 1 6 14 7 13 15
删除结点后的中序遍历:1 3 6 7 8 10 13 14 15
某些方法的非递归实现:
1. 插入节点insertNode():
//在二叉查找树中插入一个数据域为data的结点,新插入的结点一定是某个叶子节点
public TreeNode<Integer> insertNode(TreeNode<Integer> node, Integer data){
TreeNode<Integer> newNode = new TreeNode<Integer>(data,null,null);
TreeNode<Integer> tmpNode = node; //遍历节点
TreeNode<Integer> pnode = null; //记录当前节点的父节点 if(node == null){ //原树为空,新插入的记录为根节点
node = newNode;
return node;
}
while(tmpNode != null){
pnode = tmpNode;
if(tmpNode.getData() == data){ //树中存在相同关键字的结点,什么也不做
return node;
}else{
if(tmpNode.getData() > data){ //根节点>插入数据,插入到左子树中
tmpNode = tmpNode.getLchild();
}else{ //根节点<插入数据,插入到右子树中
tmpNode = tmpNode.getRchild();
}
}
}
if(pnode.getData() > data){
pnode.setLchild(newNode);
}else{
pnode.setRchild(newNode);
}
return node;
}
2. 二叉查找树的中序遍历:
//二叉查找树的中序遍历LNR,可以得到一个递增的有序数列
public void inOrder(TreeNode<Integer> node){
Stack<TreeNode<Integer>> nodeStack = new Stack<TreeNode<Integer>>();
TreeNode<Integer> tempNode = node; //遍历指针
while(tempNode != null || !nodeStack.isEmpty()){
if(tempNode != null){
nodeStack.push(tempNode);
tempNode = tempNode.getLchild();
}else{
tempNode = nodeStack.pop();
System.out.print(tempNode.getData() + " ");
tempNode = tempNode.getRchild();
}
}
}
3. 得到二叉查找树的最大值和最小值:
//查找最大值:不断地寻找右子节点
public TreeNode<Integer> getMaxData(TreeNode<Integer> node){
TreeNode<Integer> tempNode = node;
while(tempNode.getRchild()!=null){
tempNode = tempNode.getRchild();
}
return tempNode;
} //查找最小值:不断地寻找左子节点
public TreeNode<Integer> getMinData(TreeNode<Integer> node){
TreeNode<Integer> tempNode = node;
while(tempNode.getLchild() != null){
tempNode = tempNode.getLchild();
}
return tempNode;
}