【数据结构】简单实现一个二叉搜索树的基本功能（查找/插入/删除）

目录

查找

插入

删除

---

二叉搜索树是一种经典的数据结构

简单来说就是二叉搜索树若它的左子树不空，则左子树上所有结点的值均小于它的根节点的值

若它的右子树不空，则右子树上所有结点的值均大于它的根结点的值

接下来就来用代码实现三个基本操作

---

查找

思路

从二叉搜索树的定义我们可以知道

根节点的左孩子结点值小于根结点值，右孩子根节点值大于根节点值

那么我们的思路就很清晰了

如果想要找到我们想要的值，那么就从根结点开始寻找

如果这个值大于根结点的值那么就向右走，反之则向左走

代码

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

插入

思路

插入的思路和寻找的思路大体相同 

如果我们想插入这个值，那么我们就需要比较这个值是大于根节点的值还是小于根节点的值

但我们需要注意的是

如果只定义一个指针，那么就算顺利的找到了应该插入的地方

但是由于只有一个指针，所以不知道父亲结点是谁，那么就无法顺利完成插入

所以还需要定义一个指针来指向cur上次走过的地方来记录父亲结点，从而顺利插入

代码

public boolean insert(int val) {
        //如果是一颗空树，那么这个结点就是根节点
        if (root == null) {
            TreeNode node = new TreeNode(val);
            root = node;
            return true;
        }

        TreeNode parent = null;
        TreeNode cur = root;
        
        //找到应该插入的位置以及记录父亲结点的位置
        while (cur != null) {
            if (cur.val > val) {
                parent = cur;
                cur = cur.left;
            } else if (cur.val < val) {
                parent = cur;
                cur = cur.right;
            } else {
                return false;
            }
        }

        //开始插入
        TreeNode node = new TreeNode(val);
        if (parent.val < val) {
            parent.right = node;
        } else {
            parent.left = node;
        }

        return true;
    }

删除

思路

删除我们需要分情况进行三种情况来讨论

分别是：该节点左子树为空，该结点右子树为空，该节点左右子树都不为空

左子树为空

当该节点为根节点时，那么只需要将根节点定义为该节点的右孩子结点即可

          ​

当该节点不是根结点时，且父亲结点的右孩子以及拥有一个右孩子

那么直接将父亲结点的右孩子指向该结点的右孩子结点即可

                     ​

当该节点不是根结点时，且父亲结点的左孩子以及拥有一个右孩子

那么直接将父亲结点的左孩子指向该结点的右孩子结点即可​

                         

右子树为空

大体情况和左子树为空类似 

当该节点为根节点时，那么只需要将根节点定义为该节点的左孩子结点即可

                            

当该节点不是根结点时，且父亲结点的右孩子以及拥有一个左孩子

                                

 

那么直接将父亲结点的右孩子指向该结点的左孩子结点即可

                             

左右子树都不为空

当左右子树都不为空时，那么我们就时用替换法来进行删除

假如我们要删除一个左右子树都不为空的结点

那么我们首先需要找到一个数值来替换掉这个结点的数值

那么怎么找呢？我们可以找到这个结点左子树的最右结点或这右子树的最左节点

为什么这么找？因为两个值可以保证这两个值比该节点的左孩子结点大比右孩子结点小

原理是在插入的时候就已经比较过了：左子树的最右结点一定比该节点小同时比左孩子结点大

右子树的最左结点一定比该结点大但同时比右孩子小

所以这两个值的任意一个值都可以作为该节点的替换值

最后我们将这个值替换完成以后再将这个结点删除即可

但是我们在删除的时候一定要注意

有可能这个左子树的最右结点还有左孩子或者右子树的最左结点有右孩子

但是这并不影响，因为这个结点已经是最左或者最右结点了

我们在处理删除的时候只需要遵循我们上面左子树为空或者右子树为空的操作即可 

代码

 public void delete(int val){
        TreeNode cur = root;
        TreeNode parent = null;

        //首先寻找到cur值
        while (cur != null){
            if (cur.val == val){
                deleteChild(parent,cur);
                break;
            }else if (cur.val > val){
                parent = cur;
                cur = cur.left;
            }else {
                parent = cur;
                cur = cur.right;
            }

        }
    }

    //删除的操作
    public void deleteChild(TreeNode parent,TreeNode cur){
        if (cur.left == null){
            if (cur == root){
                root = cur.right;
            }else if (cur == parent.left){
                parent.left = cur.right;
            }else {
                parent.right = cur.right;
            }
        }else if (cur.right == null){
            if (cur == root){
                root = cur.left;
            }else if (cur == parent.left){
                parent.left = cur.left;
            }else {
                parent.right = cur.left;
            }
        }else{
            //这里的左右子树都不为空的删除是选取右子树的最左结点进行替换
            TreeNode target = cur.right; //替罪羊法
            TreeNode targetParent = cur;
            while (target.left != null){
                targetParent = target;
                target = targetParent.right;
            }

            cur.val = target.val;

            if (target == targetParent.left){
                targetParent.left = target.right;
            }else {
                targetParent.right = target.right;
            }
        }
    }