Java 实现二叉搜索树 代码

新建文件
创建TreeNode类,实例化

直接在BinarySearchTree类里面写就可以

java 复制代码
static class TreeNode {
        public int key;
        public TreeNode left;
        public TreeNode right;

        TreeNode(int key) {
            this.key = key;
        }
    }

    public TreeNode root;
插入节点 insert
java 复制代码
 public boolean insert(int key) {
        TreeNode node = new TreeNode(key);
        if(root==null){
            //检查是否为空树
            root=node;
            return true;//表示成功插入
        }
        TreeNode cur=root;//读取根节点
        TreeNode parent=null;
        //查找合适的位置 如果已经有了则返回false
        while(cur!=null){
            if(cur.key<key){
                parent=cur;
                cur=cur.right;
            } else if (cur.key>key) {
                parent=cur;
                cur=cur.left;
            }else{
                return false;
            }
        }
        //插入
        if(key< parent.key){
            parent.left=node;
        }else{
            parent.right=node;
        }
        return true;
    }

找到则返回该节点,没找到返回null

java 复制代码
public TreeNode search(int key) {
        if(root==null) return null;
        TreeNode cur=root;
        while(cur!=null){
            if(cur.key==key){
                return cur;
            }else if(cur.key>key){
                cur=cur.left;
            }else{
                cur=cur.right;
            }
        }
        return null;
    }
删除节点 remove

找到节点并删除,成功则返回true,失败false

java 复制代码
 public boolean remove(int key) {
        TreeNode cur=root;
        TreeNode parent=null;
        while(cur!=null){
            if(cur.key<key){
                parent=cur;
                cur=cur.right;
            } else if (cur.key>key) {
                parent=cur;
                cur=cur.left;
            }else{
                removeNode(parent,cur);//进行删除
                return true;
            }
        }
        return false;
    }

注意,这里删除又写了一个自定义函数。已至删除的位置和其双亲结点,如何让其删除呢?

如果左右子节点都在,则用左子树的最大值或者右子树的最小值替换即可,下面给出用右子树的最小值替换的方法。

java 复制代码
public  void removeNode(TreeNode parent,TreeNode cur){
        //在子树部分删除cur节点后重新构建子搜索二叉数
        if(cur.left==null){
            if(cur==root) root=cur.right;
            if(cur==parent.left) parent.left=cur.right;
            if(cur==parent.right) parent.right=cur.right;
        }else if(cur.right==null){
            if(cur==root) root=cur.left;
            if(cur==parent.left) parent.left=cur.left;
            if(cur==parent.right) parent.right=cur.left;
        }else{
            TreeNode target=cur.right;
            TreeNode targetParent=cur;
            while(target.left!=null){
                targetParent=target;
                target=target.left;
            }
            cur.key= target.key;
            if(target==targetParent.left) {
                targetParent.left = target.right;
            }else{
                targetParent.right = target.right;
            }
        }
    }

如有什么没讲清楚的地方,欢迎留言探讨!

相关推荐
逊嘘12 分钟前
【Java语言】抽象类与接口
java·开发语言·jvm
van叶~15 分钟前
算法妙妙屋-------1.递归的深邃回响:二叉树的奇妙剪枝
c++·算法
Half-up15 分钟前
C语言心型代码解析
c语言·开发语言
简简单单做算法16 分钟前
基于Retinex算法的图像去雾matlab仿真
算法·matlab·图像去雾·retinex
morris13120 分钟前
【SpringBoot】Xss的常见攻击方式与防御手段
java·spring boot·xss·csp
云卓SKYDROID30 分钟前
除草机器人算法以及技术详解!
算法·机器人·科普·高科技·云卓科技·算法技术
Source.Liu36 分钟前
【用Rust写CAD】第二章 第四节 函数
开发语言·rust
monkey_meng37 分钟前
【Rust中的迭代器】
开发语言·后端·rust
余衫马40 分钟前
Rust-Trait 特征编程
开发语言·后端·rust
monkey_meng43 分钟前
【Rust中多线程同步机制】
开发语言·redis·后端·rust