二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。二叉搜索树作为一种经典的数据结构,它既有链表的快速插入与删除操作的特点,又有数组快速查找的优势;所以应用十分广泛,例如在文件系统和数据库系统一般会采用这种数据结构进行高效率的排序与检索操作。
不论哪一种操作,所花的时间都和树的高度成正比。因此,如果共有n个元素,那么平均每次操作需要O(logn)的时间。
import java.util.Scanner;
class TreeNode {
int data;
TreeNode left, right;
public TreeNode(int data) {
this.data = data;
left = right = null;
}
}
class BinarySearchTree {
private TreeNode root;
// 插入节点
public void insert(int data) {
root = insertRec(root, data);
}
private TreeNode insertRec(TreeNode root, int data) {
if (root == null) {
root = new TreeNode(data);
return root;
}
if (data < root.data) {
root.left = insertRec(root.left, data);
} else if (data > root.data) {
root.right = insertRec(root.right, data);
}
return root;
}
// 删除节点
public void delete(int data) {
root = deleteRec(root, data);
}
private TreeNode deleteRec(TreeNode root, int data) {
if (root == null) {
return root;
}
if (data < root.data) {
root.left = deleteRec(root.left, data);
} else if (data > root.data) {
root.right = deleteRec(root.right, data);
} else {
// 节点只有一个子节点或无子节点
if (root.left == null) {
return root.right;
} else if (root.right == null) {
return root.left;
}
// 节点有两个子节点,获取中序后继
root.data = minValue(root.right);
root.right = deleteRec(root.right, root.data);
}
return root;
}
private int minValue(TreeNode root) {
int minValue = root.data;
while (root.left != null) {
minValue = root.left.data;
root = root.left;
}
return minValue;
}
// 前序遍历
public void preOrder() {
preOrderRec(root);
System.out.println();
}
private void preOrderRec(TreeNode root) {
if (root != null) {
System.out.print(root.data + " ");
preOrderRec(root.left);
preOrderRec(root.right);
}
}
// 中序遍历
public void inOrder() {
inOrderRec(root);
System.out.println();
}
private void inOrderRec(TreeNode root) {
if (root != null) {
inOrderRec(root.left);
System.out.print(root.data + " ");
inOrderRec(root.right);
}
}
// 后序遍历
public void postOrder() {
postOrderRec(root);
System.out.println();
}
private void postOrderRec(TreeNode root) {
if (root != null) {
postOrderRec(root.left);
postOrderRec(root.right);
System.out.print(root.data + " ");
}
}
}
public class BSTDemo {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
BinarySearchTree bst = new BinarySearchTree();
while (true) {
System.out.println("选择操作:");
System.out.println("1. 插入节点");
System.out.println("2. 删除节点");
System.out.println("3. 前序遍历");
System.out.println("4. 中序遍历");
System.out.println("5. 后序遍历");
System.out.println("6. 退出");
int choice = scanner.nextInt();
switch (choice) {
case 1:
System.out.print("输入要插入的整数:");
int insertData = scanner.nextInt();
bst.insert(insertData);
break;
case 2:
System.out.print("输入要删除的整数:");
int deleteData = scanner.nextInt();
bst.delete(deleteData);
break;
case 3:
System.out.println("前序遍历:");
bst.preOrder();
break;
case 4:
System.out.println("中序遍历:");
bst.inOrder();
break;
case 5:
System.out.println("后序遍历:");
bst.postOrder();
break;
case 6:
System.out.println("退出");
scanner.close();
return;
default:
System.out.println("无效选择");
}
}
}
}