二叉搜索树

目录

一，二叉搜索树概念

二，二叉搜索树的实现

2.1 insert(int key) - 插入元素

2.2 search(int key) - 查找key节点

2.3 remove(int key) - 删除key节点

---

一，二叉搜索树概念

它或者是一棵空树，或者是具有下列性质的二叉树： 若它的左子树不空，则左子树上所有结点的值均小于它的根结点的值； 若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值，且该树中没有相同的元素。如下图：

 

二，二叉搜索树的实现

public class BinarySearchTree {

    static class TreeNode {
        public int key;
        public TreeNode left;
        public TreeNode right;
        TreeNode(int key) {
            this.key = key;
        }
    }
    public TreeNode root;

    //插入一个元素
    public boolean insert(int key) {}

    //查找key是否存在
    public TreeNode search(int key) {}

    //删除key的值
    public boolean remove(int key) {}

}

2.1 insert(int key) - 插入元素

我们在二叉搜索树中的插入只能在叶子节点处，也就是说我们不会改变原来已经存储好的元素的位置，如图：

 但是我们还要注意两个细节

1.  root == null 即 树中没有元素，那么我们直接将 root = node
2.  二叉树中不能插入重复的元素，也就是说我们还要判断要插入的元素是否是重复的，如果重复，则不能插入。

接下来开始写代码：

    public boolean insert(int key) {
        TreeNode node = new TreeNode(key);//将要插入的值初始化成一个节点
        if(root == null){//处理树中没有元素的情况
            root = node;
            return true;
        }
        TreeNode cur = root;//将cur从root开始遍历
        TreeNode prev = null;//用来记录cur的父亲节点
        while(cur != null){
            prev = cur;
            if(cur.key < key){//比根节点大，往右子树移动
                cur = cur.right;
            }else if(cur.key > key){//比根节点小，往左子树移动
                cur = cur.left;
            }else{//判断是否已经含有该节点
                System.out.println("已经有该元素！");
                return false;
            }
        }
        if(prev.key < key){//判断将key插入到该节点的左边或者右边
            prev.right = node;
        }
        if(prev.key > key){
            prev.left = node;
        }
        return true;
    }

 

2.2 search(int key) - 查找key节点

非常简单，我们可以使用二叉树的遍历来解决，但是这是一棵二叉搜索树，所以更简单的做法就是利用二叉搜索树的性质（左子树<根节点<右子树），直接上代码：

    public TreeNode search(int key) {
        TreeNode cur = root;
        while(cur != null){
            if(cur.key < key){
                cur = cur.right;
            }else if(cur.key > key){
                cur = cur.left;
            }else {
                return cur;
            }
        }
        return null;
    }

2.3 remove(int key) - 删除key节点

删除方法的实现是比较困难的，如果要删除的节点是叶子节点，那么比较简单，如果不是，我们就要考虑删除该节点后其左右子树的变化，情况一下变得十分复杂。那么我们如何来实现删除的功能呢？

在这里我们先分三种情况来讨论：（cur 是要删除的节点，parent是cur的父亲节点）

1.  cur.left == null 

• cur == root 即要删除的节点就是根节点

 

• parent.left == cur 即 cur 是 parent 的左子树

• parent.right == cur 即 cur 是 parent 的右子树

2. cur.right == null

• cur == root 即 cur 就是根节点

 

• parent.left == cur 即 cur 是 parent 的左子树

• parent.right == cur 即 cur 是 parent 的右子树

3. cur.left != null && cur.right != null

•  在这里我们使用了替罪羊法来解决这一问题，我们可以将cur.left的最右边的叶子节点值覆盖cur所存的值（或者将cur.right的最左边的叶子节点值覆盖cur所存的值），然后将叶子节点删除，画个图来理解一下：

 上代码：

     public boolean remove(int key) {
        TreeNode cur = root;
        TreeNode prev = null;
        while(cur != null){//先找到要是删除的节点和其父亲节点
            if(cur.key < key){
                prev = cur;
                cur = cur.right;
            }else if(cur.key > key){
                prev = cur;
                cur = cur.left;
            }else {
                removeTreeNode(prev,cur);//删除节点
                return true;
            }
        }
        return false;
    }
    private void removeTreeNode1(TreeNode parent, TreeNode cur){
        if(cur.left == null){
            if(cur == root){
                root = cur.right;
            } else if(parent.right == cur){
                parent.right = cur.right;
            }else if(parent.left == cur){
                parent.left = cur.right;
            }
        }else if(cur.right == null){
            if(cur == root){
                root = cur.left;
            }else if(parent.right == cur){
                parent.right = cur.left;
            }else if(parent.left == cur){
                parent.left = cur.left;
            }
        }else {
            TreeNode tmpParent = cur;
            TreeNode tmpCur = cur.left;
            while(tmpCur.right != null){
                tmpParent = tmpCur;
                tmpCur = tmpCur.right;
            }
            cur.key = tmpCur.key;
            if(tmpCur == tmpParent.left){
                tmpParent.left = tmpCur.left;
            }else{
                tmpParent.right = tmpCur.left;
            }
        }
    }