二叉查找树是一种特殊的二叉树,它支持动态的数据集合的快速插入、删除和查找操作。二叉查找树的一般结构如下图所示:
从上图可以观察到二叉查找树的一个重要特性树中的任意一个节点,其左子树中的每个节点的值都小于该节点的值,右子树中的每个节点的值都大于该节点的值。
对于二叉查找树的查找操作主要步骤如下:
查找的代码如下所示:
public class BinarySearchTree { public static class Node { private int data; private Node left; private Node right;
public Node(int data) { this.data = data; } }
private Node tree;
public Node find(int data) { Node p = tree; while (p != null) { if(p.data < data) p = p.right; else if(p.data > data) p = p.left; else return p; } return null; }}
对于二叉查找树的插入操作时,新插入节点一般都是插入到叶子节点,从根节点开始依次比较新插入节点的值与当前节点值的大小关系,具体步骤如下:
插入的代码如下所示:
public class BinarySearchTree {
public static class Node { private int data; private Node left; private Node right;
public Node(int data) { this.data = data; } }
private Node tree;
public void insert(int data) { if (tree == null) { tree = new Node(data); return; }
Node p = tree; while (p != null) { if (data > p.data) { if (p.right == null) { p.right = new Node(data); return; } p = p.right; } else { // data < p.data if (p.left == null) { p.left = new Node(data); return; } p = p.left; } } }}
对于二叉查找树的删除操作主要分为三种情况,下面分别讨论:
删除的代码如下所示:
public class BinarySearchTree {
public static class Node { private int data; private Node left; private Node right;
public Node(int data) { this.data = data; } }
private Node tree;
public void delete(int data) { Node p = tree; // p 指向要删除的节点,初始化指向根节点 Node pp = null; // pp 记录的是 p 的父节点 while (p != null && p.data != data) { pp = p; if (data > p.data) p = p.right; else p = p.left; } if (p == null) return; // 没有找到
// 要删除的节点有两个子节点 if (p.left != null && p.right != null) { // 查找右子树中最小节点 Node minP = p.right; Node minPP = p; // minPP 表示 minP 的父节点 while (minP.left != null) { minPP = minP; minP = minP.left; } p.data = minP.data; // 将 minP 的数据替换到 p 中 p = minP; // 下面就变成了删除 minP 了 pp = minPP; }
// 删除节点是叶子节点或者仅有一个子节点 Node child; // p 的子节点 if (p.left != null) child = p.left; else if (p.right != null) child = p.right; else child = null;
// 删除的是根节点 if (pp == null) tree = child; else if (pp.left == p) pp.left = child; else pp.right = child; }}